From 536dbc0c9a5de27ba11e084f3c6046e1b2369397 Mon Sep 17 00:00:00 2001
From: ViperEkura <3081035982@qq.com>
Date: Tue, 7 Jul 2026 00:14:13 +0800
Subject: [PATCH] fix: set tqdm postfix before update so first step shows
 metrics

---
 astrai/trainer/train_callback.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/astrai/trainer/train_callback.py b/astrai/trainer/train_callback.py
index 28cfd1c..ec6156f 100644
--- a/astrai/trainer/train_callback.py
+++ b/astrai/trainer/train_callback.py
@@ -209,7 +209,6 @@ class ProgressBarCallback(TrainCallback):
 
     @only_on_rank(0)
     def on_optimizer_step(self, context: TrainContext):
-        self.progress_bar.update(1)
         postfix = {
             "step": context.optimizer_step,
             "loss": f"{context.loss:.4f}",
@@ -220,6 +219,7 @@ class ProgressBarCallback(TrainCallback):
         if context.val_loss is not None:
             postfix["val_loss"] = f"{context.val_loss:.4f}"
         self.progress_bar.set_postfix(postfix)
+        self.progress_bar.update(1)
 
     @only_on_rank(0)
     def on_epoch_end(self, context: TrainContext):