Skip to content

Commit 0ef07a2

Browse files
committed
update: parameters
1 parent 9301d51 commit 0ef07a2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_optimizers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def build_lookahead(*parameters, **kwargs):
104104
(build_lookahead, {'lr': 5e-1, 'weight_decay': 1e-3}, 500),
105105
(AdaBelief, {'lr': 5e-1, 'weight_decay': 1e-3}, 200),
106106
(AdaBound, {'lr': 5e-1, 'gamma': 0.1, 'weight_decay': 1e-3}, 200),
107-
(AdaBound, {'lr': 1e-2, 'gamma': 0.1, 'weight_decay': 1e-3, 'amsbound': True}, 200),
107+
(AdaBound, {'lr': 1e-1, 'gamma': 0.1, 'weight_decay': 1e-3, 'amsbound': True}, 200),
108108
(AdamP, {'lr': 5e-1, 'weight_decay': 1e-3}, 500),
109109
(DiffGrad, {'lr': 15 - 1, 'weight_decay': 1e-3}, 500),
110110
(DiffRGrad, {'lr': 1e-1, 'weight_decay': 1e-3}, 200),
@@ -124,7 +124,7 @@ def build_lookahead(*parameters, **kwargs):
124124
(AdamP, {'lr': 5e-1, 'weight_decay': 1e-3, 'adamd_debias_term': True}, 500),
125125
(DiffGrad, {'lr': 15 - 1, 'weight_decay': 1e-3, 'adamd_debias_term': True}, 500),
126126
(DiffRGrad, {'lr': 1e-1, 'weight_decay': 1e-3, 'adamd_debias_term': True}, 200),
127-
(Lamb, {'lr': 1e-1, 'weight_decay': 1e-3, 'adamd_debias_term': True}, 200),
127+
(Lamb, {'lr': 1e-1, 'weight_decay': 1e-3, 'adamd_debias_term': True}, 300),
128128
(RaLamb, {'lr': 1e-1, 'weight_decay': 1e-3, 'adamd_debias_term': True}, 500),
129129
(RAdam, {'lr': 1e-1, 'weight_decay': 1e-3, 'adamd_debias_term': True}, 200),
130130
(Ranger, {'lr': 5e-1, 'weight_decay': 1e-3, 'adamd_debias_term': True}, 200),

0 commit comments

Comments
 (0)