From 285131bfb4d05c2f15a4c22a4583bddaa2b47c7e Mon Sep 17 00:00:00 2001 From: JejuWayfarer <49282663+JejuWayfarer@users.noreply.github.com> Date: Tue, 11 Jan 2022 21:44:29 +0900 Subject: [PATCH] change metric_key_prefix in seq2seq_trainer.py (#15099) It solves the problem that metric_key_prefix is different from trainer. --- src/transformers/trainer_seq2seq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer_seq2seq.py b/src/transformers/trainer_seq2seq.py index b88fd1e09ea..799b6de7f22 100644 --- a/src/transformers/trainer_seq2seq.py +++ b/src/transformers/trainer_seq2seq.py @@ -73,7 +73,7 @@ class Seq2SeqTrainer(Trainer): self, test_dataset: Dataset, ignore_keys: Optional[List[str]] = None, - metric_key_prefix: str = "eval", + metric_key_prefix: str = "test", max_length: Optional[int] = None, num_beams: Optional[int] = None, ) -> PredictionOutput: