Fix validate_every_n_steps always running first step

kohya-ss · Jan 12, 2025 · 0456858 · 0456858
1 parent 2bbb40c
commit 0456858
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_network.py b/train_network.py
@@ -1385,6 +1385,7 @@ def remove_model(old_ckpt_name):
                 # VALIDATION PER STEP
                 should_validate_step = (
                     args.validate_every_n_steps is not None 
+                    and global_step != 0 # Skip first step
                     and global_step % args.validate_every_n_steps == 0
                 )
                 if validation_steps > 0 and should_validate_step: