remove label_names

SeanLee97 · Sep 30, 2024 · 228ae65 · 228ae65
1 parent ce52b7b
commit 228ae65
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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,