From 83bfb54f2062c87329dedfca2fa9681f5c4d043a Mon Sep 17 00:00:00 2001 From: Kohya S Date: Thu, 2 Mar 2023 19:01:22 +0900 Subject: [PATCH] fix num_repeats not working in DB classic dataset --- library/config_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/config_util.py b/library/config_util.py index d50d71f1..10961c4b 100644 --- a/library/config_util.py +++ b/library/config_util.py @@ -454,7 +454,7 @@ def generate_dreambooth_subsets_config_by_subdirs(train_data_dir: Optional[str] if num_repeats < 1: continue - subset_config = {"image_dir": str(subdir), "is_reg": is_reg, "class_tokens": class_tokens} + subset_config = {"image_dir": str(subdir), "num_repeats": num_repeats, "is_reg": is_reg, "class_tokens": class_tokens} subsets_config.append(subset_config) return subsets_config