diff options
Diffstat (limited to 'dreambooth.py')
-rw-r--r-- | dreambooth.py | 8 |
1 files changed, 4 insertions, 4 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( |