Merge branch 'main' of https://github.com/allenai/LLM

allenai · Jul 18, 2024 · 345edc6 · 345edc6
2 parents 66d2be7 + 37756a7
commit 345edc6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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 (