summaryrefslogtreecommitdiffstats
path: root/train_ti.py
diff options
context:
space:
mode:
Diffstat (limited to 'train_ti.py')
-rw-r--r--train_ti.py4
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