diff --git a/train_network.py b/train_network.py
index a7e747c6..f66cdeb4 100644
--- a/train_network.py
+++ b/train_network.py
@@ -171,10 +171,9 @@ class NetworkTrainer:
         train_dataset_group: Union[train_util.DatasetGroup, train_util.MinimalDataset],
         val_dataset_group: Optional[train_util.DatasetGroup],
     ):
-        if type(self).__name__ == "NetworkTrainer":
-            train_dataset_group.verify_bucket_reso_steps(64)
-            if val_dataset_group is not None:
-                val_dataset_group.verify_bucket_reso_steps(64)
+        train_dataset_group.verify_bucket_reso_steps(64)
+        if val_dataset_group is not None:
+            val_dataset_group.verify_bucket_reso_steps(64)
 
     def load_target_model(self, args, weight_dtype, accelerator):
         text_encoder, vae, unet, _ = train_util.load_target_model(args, weight_dtype, accelerator)