From 0ed9290c312d125ff73876056703df32a48c748a Mon Sep 17 00:00:00 2001
From: Shivam Mehta <shivam.mehta25@gmail.com>
Date: Wed, 6 Dec 2023 10:39:54 +0000
Subject: [PATCH] Logging global step while training

---
 matcha/models/baselightningmodule.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/matcha/models/baselightningmodule.py b/matcha/models/baselightningmodule.py
index 29f4927..3724888 100644
--- 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,
         )