riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix typo in trainer.py (#31048)

Browse Source
This commit is contained in:
Sina Taslimi 2024-05-28 13:09:32 +02:00 committed by GitHub
parent 98e2d48e9a
commit 66add161dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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.py
View File

@ -3565,7 +3565,7 @@ class Trainer:
When used with `load_best_model_at_end`, make sure `metric_for_best_model` references exactly one 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 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 `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`.
</Tip> </Tip>