Logging global step while training

2026-02-04 09:49:21 +08:00 · 2023-12-06 10:39:54 +00:00
parent f39ee6cf3b
commit 0ed9290c31
1 changed files with 1 additions and 1 deletions
--- a/matcha/models/baselightningmodule.py
+++ b/matcha/models/baselightningmodule.py
@@ -81,7 +81,7 @@ class BaseLightningClass(LightningModule, ABC):
            "step",
            float(self.global_step),
            on_step=True,
-            on_epoch=True,
+            prog_bar=True,
            logger=True,
            sync_dist=True,
        )