From 6c47c433f21d62b274d7aff82be767847b8090b1 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 16 Apr 2023 19:17:27 +0200 Subject: Fix --- training/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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() -- cgit v1.2.3-70-g09d2