From d7470b845caaf6fdf452fc48274339102bc90778 Mon Sep 17 00:00:00 2001
From: Liyu Xiao <liyu@qoherent.ai>
Date: Mon, 26 May 2025 15:02:57 -0400
Subject: [PATCH] increased the batch size for every update

---
 data/training/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/data/training/train.py b/data/training/train.py
index ad7015b..314d1ba 100644
--- a/data/training/train.py
+++ b/data/training/train.py
@@ -17,7 +17,7 @@ from lightning.pytorch.callbacks import TQDMProgressBar
 
 class CustomProgressBar(TQDMProgressBar):
     def __init__(self):
-        super().__init__(refresh_rate=1)  # update every batch
+        super().__init__(refresh_rate=128)  # update every batch
 
 
 def train_model():