From 6fff6cc33e62e69d79edf6184c81da6fb917cde4 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Mon, 26 May 2025 02:10:48 +0800 Subject: [PATCH] Update train_network.py --- train_network.py | 1 - 1 file changed, 1 deletion(-) diff --git a/train_network.py b/train_network.py index a42317a7..26668b97 100644 --- a/train_network.py +++ b/train_network.py @@ -858,7 +858,6 @@ class NetworkTrainer: ) loss_recorder = train_util.LossRecorder() - del train_dataset_group # callback for step start if hasattr(accelerator.unwrap_model(network), "on_step_start"):