diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index b706d07..f8f6e84 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
@@ -287,8 +287,8 @@ def parse_args(): | |||
287 | parser.add_argument( | 287 | parser.add_argument( |
288 | "--optimizer", | 288 | "--optimizer", |
289 | type=str, | 289 | type=str, |
290 | default="adam", | 290 | default="dadan", |
291 | help='Optimizer to use ["adam", "adam8bit"]' | 291 | help='Optimizer to use ["adam", "adam8bit", "dadam", "dadan"]' |
292 | ) | 292 | ) |
293 | parser.add_argument( | 293 | parser.add_argument( |
294 | "--adam_beta1", | 294 | "--adam_beta1", |