riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

[FIX] Save speed metrics to logs (#38136)

Browse Source 
Previously, we calculated speed metrics and did not do anything with the result.
This commit is contained in:
Pavel Gein 2025-05-15 19:58:50 +05:00 committed by GitHub
parent 27ef46e846
commit 0f77ca72ca
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

@ -3653,7 +3653,7 @@ class Trainer:
if self.args.include_num_input_tokens_seen:
logs["num_input_tokens_seen"] = self.state.num_input_tokens_seen
if start_time is not None:
speed_metrics("train", start_time, num_tokens=self.state.num_input_tokens_seen)
logs.update(speed_metrics("train", start_time, num_tokens=self.state.num_input_tokens_seen))
output = {**logs, **{"step": self.state.global_step}}
self.state.log_history.append(output)