SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Trainer] Report both steps and num samples per second (#11818)

Browse Source 
* [Trainer] Report both steps and num samples per second

* Fix batch number

* Update src/transformers/trainer_utils.py

Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>

* Address review comments

Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>
This commit is contained in:
Sylvain Gugger
2021-05-24 19:51:42 -04:00
committed by GitHub
parent eaab9397cd
commit afe479adb5
5 changed files with 36 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_trainer.py
View File

@@ -316,6 +316,8 @@ class TrainerIntegrationCommon:
_ = log1.pop("train_runtime", None)
_ = log.pop("train_samples_per_second", None)
_ = log1.pop("train_samples_per_second", None)
_ = log.pop("train_steps_per_second", None)
_ = log1.pop("train_steps_per_second", None)
self.assertEqual(log, log1)