Update train_network.py

This commit is contained in:
DKnight54
2025-05-26 02:10:48 +08:00
committed by GitHub
parent 48e76c8aab
commit 6fff6cc33e

View File

@@ -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"):