From 228ae65d75930b3069e39e24eee141db17263b16 Mon Sep 17 00:00:00 2001 From: Sean Lee Date: Mon, 30 Sep 2024 09:15:05 +0800 Subject: [PATCH] remove label_names --- angle_emb/angle.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/angle_emb/angle.py b/angle_emb/angle.py index a2d55e5..c5a60e6 100644 --- a/angle_emb/angle.py +++ b/angle_emb/angle.py @@ -857,7 +857,6 @@ def compute_loss(self, model, inputs, return_outputs=False): if mask_target_labels is not None: loss += self.compute_mlm_loss(mlm_logits, mask_target_labels) - inputs['labels'] = None # avoid evaluation error return (loss, outputs) if return_outputs else loss @@ -990,7 +989,6 @@ def compute_loss(self, model, inputs, return_outputs=False): if mask_target_labels is not None: loss += self.compute_mlm_loss(mlm_logits, mask_target_labels) - inputs['labels'] = None # avoid evaluation error return (loss, teacher_outputs) if return_outputs else loss @@ -1547,7 +1545,6 @@ def fit(self, save_total_limit=save_total_limit, load_best_model_at_end=False, ddp_find_unused_parameters=False if self.gpu_count > 1 else None, - label_names=AnglE.special_columns, **argument_kwargs, ), callbacks=callbacks,