diff options
Diffstat (limited to 'train_ti.py')
-rw-r--r-- | train_ti.py | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/train_ti.py b/train_ti.py index 036c288..7aeff7c 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -620,8 +620,6 @@ def main(): | |||
620 | eps=args.adam_epsilon, | 620 | eps=args.adam_epsilon, |
621 | decouple=True, | 621 | decouple=True, |
622 | ) | 622 | ) |
623 | |||
624 | args.learning_rate = 1.0 | ||
625 | elif args.optimizer == 'dadan': | 623 | elif args.optimizer == 'dadan': |
626 | try: | 624 | try: |
627 | import dadaptation | 625 | import dadaptation |
@@ -633,8 +631,6 @@ def main(): | |||
633 | weight_decay=args.adam_weight_decay, | 631 | weight_decay=args.adam_weight_decay, |
634 | eps=args.adam_epsilon, | 632 | eps=args.adam_epsilon, |
635 | ) | 633 | ) |
636 | |||
637 | args.learning_rate = 1.0 | ||
638 | else: | 634 | else: |
639 | raise ValueError(f"Unknown --optimizer \"{args.optimizer}\"") | 635 | raise ValueError(f"Unknown --optimizer \"{args.optimizer}\"") |
640 | 636 | ||