diff --git a/src/llamafactory/train/callbacks.py b/src/llamafactory/train/callbacks.py index 3b05317d..b15db4ef 100644 --- a/src/llamafactory/train/callbacks.py +++ b/src/llamafactory/train/callbacks.py @@ -302,6 +302,7 @@ class LogCallback(TrainerCallback): learning_rate=state.log_history[-1].get("learning_rate", None), epoch=state.log_history[-1].get("epoch", None), percentage=round(self.cur_steps / self.max_steps * 100, 2) if self.max_steps != 0 else 100, + cur_time = time.time(), elapsed_time=self.elapsed_time, remaining_time=self.remaining_time, throughput="{:.2f}".format(state.num_input_tokens_seen / (time.time() - self.start_time)),