diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index 4456bd1..48b7926 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
@@ -539,7 +539,6 @@ def main(): | |||
539 | elif args.optimizer == 'adafactor': | 539 | elif args.optimizer == 'adafactor': |
540 | create_optimizer = partial( | 540 | create_optimizer = partial( |
541 | transformers.optimization.Adafactor, | 541 | transformers.optimization.Adafactor, |
542 | beta1=args.adam_beta1, | ||
543 | weight_decay=args.adam_weight_decay, | 542 | weight_decay=args.adam_weight_decay, |
544 | scale_parameter=True, | 543 | scale_parameter=True, |
545 | relative_step=True, | 544 | relative_step=True, |