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(
     try:
         for epoch in range(num_epochs):
             if accelerator.is_main_process:
-                if epoch % sample_frequency == 0 and (cycle == 1 or epoch != 0):
+                if epoch % sample_frequency == 0 and (cycle == 0 or epoch != 0):
                     local_progress_bar.clear()
                     global_progress_bar.clear()