diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 5 |
1 files changed, 2 insertions, 3 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(): | |||
133 | help="Tag dropout probability.", | 133 | help="Tag dropout probability.", |
134 | ) | 134 | ) |
135 | parser.add_argument( | 135 | parser.add_argument( |
136 | "--tag_shuffle", | 136 | "--no_tag_shuffle", |
137 | action="store_true", | 137 | action="store_true", |
138 | default=True, | ||
139 | help="Shuffle tags.", | 138 | help="Shuffle tags.", |
140 | ) | 139 | ) |
141 | parser.add_argument( | 140 | parser.add_argument( |
@@ -774,7 +773,7 @@ def main(): | |||
774 | bucket_step_size=args.bucket_step_size, | 773 | bucket_step_size=args.bucket_step_size, |
775 | bucket_max_pixels=args.bucket_max_pixels, | 774 | bucket_max_pixels=args.bucket_max_pixels, |
776 | dropout=args.tag_dropout, | 775 | dropout=args.tag_dropout, |
777 | shuffle=args.tag_shuffle, | 776 | shuffle=not args.no_tag_shuffle, |
778 | template_key=args.train_data_template, | 777 | template_key=args.train_data_template, |
779 | valid_set_size=args.valid_set_size, | 778 | valid_set_size=args.valid_set_size, |
780 | valid_set_repeat=args.valid_set_repeat, | 779 | valid_set_repeat=args.valid_set_repeat, |