From d0ce16b542deac464e097c38adc5095802bd6763 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 3 Oct 2022 11:44:42 +0200 Subject: Assign unused images in validation dataset to train dataset --- textual_inversion.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'textual_inversion.py') diff --git a/textual_inversion.py b/textual_inversion.py index 7a7d7fc..fa6214e 100644 --- a/textual_inversion.py +++ b/textual_inversion.py @@ -604,9 +604,15 @@ def main(): ) datamodule = CSVDataModule( - data_file=args.train_data_file, batch_size=args.train_batch_size, tokenizer=tokenizer, - size=args.resolution, placeholder_token=args.placeholder_token, repeats=args.repeats, - center_crop=args.center_crop) + data_file=args.train_data_file, + batch_size=args.train_batch_size, + tokenizer=tokenizer, + size=args.resolution, + placeholder_token=args.placeholder_token, + repeats=args.repeats, + center_crop=args.center_crop, + valid_set_size=args.sample_batch_size*args.stable_sample_batches + ) datamodule.prepare_data() datamodule.setup() -- cgit v1.2.3-54-g00ecf