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,