Skip the final 1 step

2026-04-08 06:28:48 +00:00 · 2024-05-31 12:24:15 +08:00
parent b2363f1021
commit 3eb27ced52
1 changed files with 4 additions and 0 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -897,6 +897,10 @@ class NetworkTrainer:

            for step, batch in enumerate(skipped_dataloader or train_dataloader):
                current_step.value = global_step
+                if initial_step > 0:
+                    initial_step -= 1
+                    continue
+
                with accelerator.accumulate(training_model):
                    on_step_start(text_encoder, unet)