From b57ca669a150d9313447612fb8c37668f4f2a80d Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 9 Jan 2023 10:19:37 +0100 Subject: Add --valid_set_repeat --- train_dreambooth.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index d396249..aa5ff01 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -383,6 +383,12 @@ def parse_args(): default=None, help="Number of images in the validation dataset." ) + parser.add_argument( + "--valid_set_repeat", + type=int, + default=None, + help="Times the images in the validation dataset are repeated." + ) parser.add_argument( "--train_batch_size", type=int, @@ -451,6 +457,9 @@ def parse_args(): if isinstance(args.exclude_collections, str): args.exclude_collections = [args.exclude_collections] + if args.valid_set_repeat is None: + args.valid_set_repeat = args.train_batch_size + if args.output_dir is None: raise ValueError("You must specify --output_dir") @@ -764,6 +773,7 @@ def main(): dropout=args.tag_dropout, template_key=args.train_data_template, valid_set_size=args.valid_set_size, + valid_set_repeat=args.valid_set_repeat, num_workers=args.dataloader_num_workers, seed=args.seed, filter=keyword_filter, -- cgit v1.2.3-54-g00ecf