Undo train_network change

This commit is contained in:
rockerBOO
2025-04-05 19:06:32 -04:00
parent 0392a57210
commit ef3a110ae1

View File

@@ -171,7 +171,6 @@ class NetworkTrainer:
train_dataset_group: Union[train_util.DatasetGroup, train_util.MinimalDataset], train_dataset_group: Union[train_util.DatasetGroup, train_util.MinimalDataset],
val_dataset_group: Optional[train_util.DatasetGroup], val_dataset_group: Optional[train_util.DatasetGroup],
): ):
if type(self).__name__ == "NetworkTrainer":
train_dataset_group.verify_bucket_reso_steps(64) train_dataset_group.verify_bucket_reso_steps(64)
if val_dataset_group is not None: if val_dataset_group is not None:
val_dataset_group.verify_bucket_reso_steps(64) val_dataset_group.verify_bucket_reso_steps(64)