fix: resume optimizer LR, step display, and consumed_samples alignment

2026-07-15 08:59:52 +08:00
parent 13998da15a
commit bb175fda91
3 changed files with 8 additions and 2 deletions
@@ -210,7 +210,7 @@ class ProgressBarCallback(TrainCallback):
    @only_on_rank(0)
    def on_optimizer_step(self, context: TrainContext):
        postfix = {
-            "step": context.optimizer_step,
+            "step": f"{context.optimizer_step:d}",
            "loss": f"{context.loss:.4f}",
            "lr": f"{context.optimizer.param_groups[-1]['lr']:.2e}",
        }