VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-18 01:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Disable repeats for validation datasets

Browse Source
This commit is contained in:
rockerBOO
2025-01-12 16:42:04 -05:00
parent 25929dd0d7
commit b489082495
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
library/train_util.py
View File

@@ -2055,9 +2055,10 @@ class DreamBoothDataset(BaseDataset):
num_reg_images = 0 num_reg_images = 0
reg_infos: List[Tuple[ImageInfo, DreamBoothSubset]] = [] reg_infos: List[Tuple[ImageInfo, DreamBoothSubset]] = []
for subset in subsets: for subset in subsets:
if subset.num_repeats < 1: num_repeats = subset.num_repeats if self.is_training_dataset else 1
if num_repeats < 1:
logger.warning( logger.warning(
f"ignore subset with image_dir='{subset.image_dir}': num_repeats is less than 1 / num_repeatsが1を下回っているためサブセットを無視します: {subset.num_repeats}" f"ignore subset with image_dir='{subset.image_dir}': num_repeats is less than 1 / num_repeatsが1を下回っているためサブセットを無視します: {num_repeats}"
) )
continue continue
@@ -2075,12 +2076,12 @@ class DreamBoothDataset(BaseDataset):
continue continue
if subset.is_reg: if subset.is_reg:
num_reg_images += subset.num_repeats * len(img_paths) num_reg_images += num_repeats * len(img_paths)
else: else:
num_train_images += subset.num_repeats * len(img_paths) num_train_images += num_repeats * len(img_paths)
for img_path, caption, size in zip(img_paths, captions, sizes): for img_path, caption, size in zip(img_paths, captions, sizes):
info = ImageInfo(img_path, subset.num_repeats, caption, subset.is_reg, img_path) info = ImageInfo(img_path, num_repeats, caption, subset.is_reg, img_path)
if size is not None: if size is not None:
info.image_size = size info.image_size = size
if subset.is_reg: if subset.is_reg: