From 0f77ca72cae3565632bafd7e06080b2c19920f06 Mon Sep 17 00:00:00 2001 From: Pavel Gein Date: Thu, 15 May 2025 19:58:50 +0500 Subject: [PATCH] [FIX] Save speed metrics to logs (#38136) Previously, we calculated speed metrics and did not do anything with the result. --- src/transformers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index 5886146002f..586a5335c0b 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -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)