riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

change metric_key_prefix in seq2seq_trainer.py (#15099)

Browse Source 
It solves the problem that metric_key_prefix is different from trainer.
This commit is contained in:
JejuWayfarer 2022-01-11 21:44:29 +09:00 committed by GitHub
parent c4fa908fa9
commit 285131bfb4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/trainer_seq2seq.py
View File

@ -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: