diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-10 10:48:53 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-10 10:48:53 +0100 |
| commit | e24ba7d0a81e9a268c95b20ece75dd74a39fa75f (patch) | |
| tree | ffc920e7d62028f81d50f80d5a9a15efc4c3e32a /train_dreambooth.py | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-e24ba7d0a81e9a268c95b20ece75dd74a39fa75f.tar.gz textual-inversion-diff-e24ba7d0a81e9a268c95b20ece75dd74a39fa75f.tar.bz2 textual-inversion-diff-e24ba7d0a81e9a268c95b20ece75dd74a39fa75f.zip | |
Better defaults
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, |