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