summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--dreambooth.py8
-rw-r--r--textual_inversion.py6
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():
68 parser.add_argument( 68 parser.add_argument(
69 "--num_class_images", 69 "--num_class_images",
70 type=int, 70 type=int,
71 default=200, 71 default=400,
72 help="How many class images to generate per training image." 72 help="How many class images to generate per training image."
73 ) 73 )
74 parser.add_argument( 74 parser.add_argument(
@@ -128,7 +128,7 @@ def parse_args():
128 parser.add_argument( 128 parser.add_argument(
129 "--learning_rate", 129 "--learning_rate",
130 type=float, 130 type=float,
131 default=1e-6, 131 default=1e-7,
132 help="Initial learning rate (after the potential warmup period) to use.", 132 help="Initial learning rate (after the potential warmup period) to use.",
133 ) 133 )
134 parser.add_argument( 134 parser.add_argument(
@@ -149,7 +149,7 @@ def parse_args():
149 parser.add_argument( 149 parser.add_argument(
150 "--lr_warmup_steps", 150 "--lr_warmup_steps",
151 type=int, 151 type=int,
152 default=0, 152 default=200,
153 help="Number of steps for the warmup in the lr scheduler." 153 help="Number of steps for the warmup in the lr scheduler."
154 ) 154 )
155 parser.add_argument( 155 parser.add_argument(
@@ -172,7 +172,7 @@ def parse_args():
172 parser.add_argument( 172 parser.add_argument(
173 "--adam_weight_decay", 173 "--adam_weight_decay",
174 type=float, 174 type=float,
175 default=1e-2, 175 default=0,
176 help="Weight decay to use." 176 help="Weight decay to use."
177 ) 177 )
178 parser.add_argument( 178 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():
69 parser.add_argument( 69 parser.add_argument(
70 "--num_class_images", 70 "--num_class_images",
71 type=int, 71 type=int,
72 default=200, 72 default=400,
73 help="How many class images to generate per training image." 73 help="How many class images to generate per training image."
74 ) 74 )
75 parser.add_argument( 75 parser.add_argument(
76 "--repeats", 76 "--repeats",
77 type=int, 77 type=int,
78 default=100, 78 default=1,
79 help="How many times to repeat the training data." 79 help="How many times to repeat the training data."
80 ) 80 )
81 parser.add_argument( 81 parser.add_argument(
@@ -149,7 +149,7 @@ def parse_args():
149 parser.add_argument( 149 parser.add_argument(
150 "--lr_warmup_steps", 150 "--lr_warmup_steps",
151 type=int, 151 type=int,
152 default=500, 152 default=200,
153 help="Number of steps for the warmup in the lr scheduler." 153 help="Number of steps for the warmup in the lr scheduler."
154 ) 154 )
155 parser.add_argument( 155 parser.add_argument(