Adding modified train_util and config_util

This commit is contained in:
rockerBOO
2025-01-02 15:59:20 -05:00
parent 8743532963
commit 449c1c5c50
2 changed files with 1 additions and 2 deletions

View File

@@ -84,7 +84,6 @@ class BaseDatasetParams:
tokenizer: Union[CLIPTokenizer, List[CLIPTokenizer]] = None
max_token_length: int = None
resolution: Optional[Tuple[int, int]] = None
network_multiplier: float = 1.0
debug_dataset: bool = False
validation_seed: Optional[int] = None
validation_split: float = 0.0

View File

@@ -1420,7 +1420,7 @@ class DreamBoothDataset(BaseDataset):
img_paths = glob_images(subset.image_dir, "*")
if self.validation_split > 0.0:
img_paths = split_train_val(img_paths, self.is_train, self.validation_split, self.validation_seed)
logger.info(f"found directory {subset.image_dir} contains {len(img_paths)} image files")
print(f"found directory {subset.image_dir} contains {len(img_paths)} image files")
# 画像ファイルごとにプロンプトを読み込み、もしあればそちらを使う
captions = []