diff options
author | Volpeon <git@volpeon.ink> | 2022-10-10 07:54:43 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2022-10-10 07:54:43 +0200 |
commit | 9a23c86587d276ab83244c9d78b3667991039398 (patch) | |
tree | 21d4f084fd389c3f1dd9d73c687c3caf0f5e6747 | |
parent | Update (diff) | |
download | textual-inversion-diff-9a23c86587d276ab83244c9d78b3667991039398.tar.gz textual-inversion-diff-9a23c86587d276ab83244c9d78b3667991039398.tar.bz2 textual-inversion-diff-9a23c86587d276ab83244c9d78b3667991039398.zip |
Updated default params
-rw-r--r-- | dreambooth.py | 8 | ||||
-rw-r--r-- | 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(): | |||
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( |