Clear sizes for validation reg images to be consistent

2026-04-08 22:35:09 +00:00 · 2025-02-16 22:28:34 -05:00
parent 3c7496ae3f
commit f3a010978c
1 changed files with 1 additions and 0 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1990,6 +1990,7 @@ class DreamBoothDataset(BaseDataset):
                    # Skip any validation dataset for regularization images
                    if self.is_training_dataset is False:
                        img_paths = []
+                        sizes = []
                    # Otherwise the img_paths remain as original img_paths and no split 
                    # required for training images dataset of regularization images
                else: