diff --git a/olmo/train.py b/olmo/train.py index 1d4143c7c..db6f15045 100644 --- a/olmo/train.py +++ b/olmo/train.py @@ -1154,9 +1154,12 @@ def on_trace_ready(p): # Log metrics to console. if self.global_step % self.cfg.console_log_interval == 0: if get_global_rank() == 0: - self.log_metrics_to_console(f"[step={self.global_step}/{self.max_steps}]", metrics) + self.log_metrics_to_console( + f"[step={self.global_step}/{self.max_steps},epoch={epoch}/{self.max_epochs}]", + metrics, + ) else: - log.info(f"[step={self.global_step}/{self.max_steps}]") + log.info(f"[step={self.global_step}/{self.max_steps},epoch={epoch}/{self.max_epochs}]") # Log metrics to W&B. if (