From e24ba7d0a81e9a268c95b20ece75dd74a39fa75f Mon Sep 17 00:00:00 2001 From: Volpeon Date: Tue, 10 Jan 2023 10:48:53 +0100 Subject: Better defaults --- train_dreambooth.py | 5 ++--- train_ti.py | 8 ++++---- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/train_dreambooth.py b/train_dreambooth.py index 21fe2fb..0182693 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -133,9 +133,8 @@ def parse_args(): help="Tag dropout probability.", ) parser.add_argument( - "--tag_shuffle", + "--no_tag_shuffle", action="store_true", - default=True, help="Shuffle tags.", ) parser.add_argument( @@ -774,7 +773,7 @@ def main(): bucket_step_size=args.bucket_step_size, bucket_max_pixels=args.bucket_max_pixels, dropout=args.tag_dropout, - shuffle=args.tag_shuffle, + shuffle=not args.no_tag_shuffle, template_key=args.train_data_template, valid_set_size=args.valid_set_size, valid_set_repeat=args.valid_set_repeat, diff --git a/train_ti.py b/train_ti.py index b88ccc3..4e2c3c5 100644 --- a/train_ti.py +++ b/train_ti.py @@ -165,18 +165,18 @@ def parse_args(): parser.add_argument( "--tag_dropout", type=float, - default=0.1, + default=0, help="Tag dropout probability.", ) parser.add_argument( - "--tag_shuffle", + "--no_tag_shuffle", action="store_true", help="Shuffle tags.", ) parser.add_argument( "--vector_dropout", type=int, - default=0, + default=0.1, help="Vector dropout probability.", ) parser.add_argument( @@ -750,7 +750,7 @@ def main(): bucket_step_size=args.bucket_step_size, bucket_max_pixels=args.bucket_max_pixels, dropout=args.tag_dropout, - shuffle=args.tag_shuffle, + shuffle=not args.no_tag_shuffle, template_key=args.train_data_template, valid_set_size=args.valid_set_size, valid_set_repeat=args.valid_set_repeat, -- cgit v1.2.3-70-g09d2