dont compute metrics

SeanLee97 · Sep 29, 2024 · cfa4f52 · cfa4f52
1 parent 271384f
commit cfa4f52
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/angle_emb/angle.py b/angle_emb/angle.py
@@ -1526,6 +1526,7 @@ def fit(self,
             dataset_format=self.detect_dataset_format(train_ds),
             train_dataset=train_ds,
             eval_dataset=valid_ds,
+            compute_metrics=None,  # do not compute metrics
             loss_kwargs=loss_kwargs,
             tokenizer=self.tokenizer,
             args=TrainingArguments(