qoherent/modrec-workflow
14
0
Fork 1
Code Issues 1 Pull Requests Workflows Packages Projects Releases 1 Wiki Activity

increased the batch size for every update
All checks were successful
RIA Hub Workflow Demo / ria-demo (push) Successful in 3m40s
Details

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): class CustomProgressBar(TQDMProgressBar):
def __init__(self): def __init__(self):
super().__init__(refresh_rate=1) # update every batch super().__init__(refresh_rate=128) # update every batch
def train_model(): def train_model():