michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

increased the batch size for every update

Browse Source
This commit is contained in:
Liyu Xiao 2025-05-26 15:02:57 -04:00
parent 2a4ecd175a
commit d7470b845c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data/training/train.py
View File

@ -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():