From 9a23c86587d276ab83244c9d78b3667991039398 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Mon, 10 Oct 2022 07:54:43 +0200
Subject: Updated default params

---
 dreambooth.py        | 8 ++++----
 textual_inversion.py | 6 +++---
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/dreambooth.py b/dreambooth.py
index 48fc7f2..9f1b7af 100644
--- a/dreambooth.py
+++ b/dreambooth.py
@@ -68,7 +68,7 @@ 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(
@@ -128,7 +128,7 @@ def parse_args():
     parser.add_argument(
         "--learning_rate",
         type=float,
-        default=1e-6,
+        default=1e-7,
         help="Initial learning rate (after the potential warmup period) to use.",
     )
     parser.add_argument(
@@ -149,7 +149,7 @@ def parse_args():
     parser.add_argument(
         "--lr_warmup_steps",
         type=int,
-        default=0,
+        default=200,
         help="Number of steps for the warmup in the lr scheduler."
     )
     parser.add_argument(
@@ -172,7 +172,7 @@ def parse_args():
     parser.add_argument(
         "--adam_weight_decay",
         type=float,
-        default=1e-2,
+        default=0,
         help="Weight decay to use."
     )
     parser.add_argument(
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(
-- 
cgit v1.2.3-70-g09d2