From cfa4f52999b06a26aa5734509b6c3ae3e446148a Mon Sep 17 00:00:00 2001 From: Sean Lee Date: Sun, 29 Sep 2024 20:47:11 +0800 Subject: [PATCH] dont compute metrics --- angle_emb/angle.py | 1 + 1 file changed, 1 insertion(+) diff --git a/angle_emb/angle.py b/angle_emb/angle.py index 8466425..7640b17 100644 --- 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(