Updated default params

1 files changed, 3 insertions, 3 deletions

diff --git a/textual_inversion.py b/textual_inversion.py
index e641cab..7373472 100644
--- a/textual_inversion.py
+++ b/textual_inversion.py
@@ -69,13 +69,13 @@ def parse_args():
     parser.add_argument(
         "--num_class_images",
         type=int,
-        default=200,
+        default=400,
         help="How many class images to generate per training image."
     )
     parser.add_argument(
         "--repeats",
         type=int,
-        default=100,
+        default=1,
         help="How many times to repeat the training data."
     )
     parser.add_argument(
@@ -149,7 +149,7 @@ def parse_args():
     parser.add_argument(
         "--lr_warmup_steps",
         type=int,
-        default=500,
+        default=200,
         help="Number of steps for the warmup in the lr scheduler."
     )
     parser.add_argument(