hard code to fix evaluation error

SeanLee97 · Sep 29, 2024 · ce52b7b · ce52b7b
1 parent daf2bb6
commit ce52b7b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/angle_emb/angle.py b/angle_emb/angle.py
@@ -857,6 +857,7 @@ 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
 
 
@@ -989,6 +990,7 @@ 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