feat: only update on 100 iters

AshishKumar4 · Aug 5, 2024 · 3fd4185 · 3fd4185
1 parent 5e16bd7
commit 3fd4185
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/training.py b/training.py
@@ -653,9 +653,9 @@ def train_loop(current_epoch, pbar: tqdm.tqdm, train_state, rng_state):
                 epoch_loss += loss
 
                 if pbar is not None:
-                    if i % 10 == 0:
+                    if i % 100 == 0:
                         pbar.set_postfix(loss=f'{loss:.4f}')
-                        pbar.update(10)
+                        pbar.update(100)
                         current_step = current_epoch*steps_per_epoch + i
                         if self.wandb is not None:
                             self.wandb.log({