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(