diff options
Diffstat (limited to 'training')
-rw-r--r-- | training/functional.py | 2 |
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 | ||