summaryrefslogtreecommitdiffstats
path: root/training
diff options
context:
space:
mode:
Diffstat (limited to 'training')
-rw-r--r--training/functional.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/functional.py b/training/functional.py
index ebc40de..c6ceb20 100644
--- a/training/functional.py
+++ b/training/functional.py
@@ -519,7 +519,7 @@ def train_loop(
519 try: 519 try:
520 for epoch in range(num_epochs): 520 for epoch in range(num_epochs):
521 if accelerator.is_main_process: 521 if accelerator.is_main_process:
522 if epoch % sample_frequency == 0 and (cycle == 1 or epoch != 0): 522 if epoch % sample_frequency == 0 and (cycle == 0 or epoch != 0):
523 local_progress_bar.clear() 523 local_progress_bar.clear()
524 global_progress_bar.clear() 524 global_progress_bar.clear()
525 525