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