Merge 15d84bcfb4 into 1dae34b0af

2026-04-10 15:00:23 +00:00 · 2026-04-01 15:38:45 +00:00
parent 1dae34b0af 15d84bcfb4
commit e5823d6271
1 changed files with 4 additions and 0 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -1478,6 +1478,8 @@ class NetworkTrainer:
                    )
                    progress_bar.unpause()

+                    clean_memory_on_device(accelerator.device)
+
                    # 指定ステップごとにモデルを保存
                    if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0:
                        accelerator.wait_for_everyone()
@@ -1693,6 +1695,8 @@ class NetworkTrainer:
            progress_bar.unpause()
            optimizer_train_fn()

+            clean_memory_on_device(accelerator.device)
+
            # end of epoch

        # metadata["ss_epoch"] = str(num_train_epochs)