From 66add161dcaa91f4e60c8e3224ed297ab72e7b0f Mon Sep 17 00:00:00 2001 From: Sina Taslimi <33656391+taslimisina@users.noreply.github.com> Date: Tue, 28 May 2024 13:09:32 +0200 Subject: [PATCH] Fix typo in trainer.py (#31048) --- src/transformers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index 20c709b248a..58e5fd14b6f 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -3565,7 +3565,7 @@ def evaluate( When used with `load_best_model_at_end`, make sure `metric_for_best_model` references exactly one of the datasets. If you, for example, pass in `{"data1": data1, "data2": data2}` for two datasets `data1` and `data2`, you could specify `metric_for_best_model="eval_data1_loss"` for using the - loss on `data1` and `metric_for_best_model="eval_data1_loss"` for the loss on `data2`. + loss on `data1` and `metric_for_best_model="eval_data2_loss"` for the loss on `data2`.