Skip to content

Commit 534c200

Browse files
committed
update: OPTIMIZERS
1 parent 9d4a1a4 commit 534c200

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test_optimizers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,7 @@ def build_lookahead(*parameters, **kwargs):
102102
(DiffGrad, {'lr': 5e-1, 'weight_decay': 1e-3}, 200),
103103
(DiffRGrad, {'lr': 5e-1, 'weight_decay': 1e-3}, 200),
104104
(Lamb, {'lr': 1e-1, 'weight_decay': 1e-3}, 500),
105+
(Lamb, {'lr': 1e-1, 'weight_decay': 1e-3, 'adam': True, 'eps': 1e-8}, 500),
105106
(Lamb, {'lr': 1e-1, 'weight_decay': 1e-3, 'pre_norm': True, 'eps': 1e-8}, 500),
106107
(LARS, {'lr': 1e-1, 'weight_decay': 1e-3}, 500),
107108
(RaLamb, {'lr': 1e-1, 'weight_decay': 1e-3}, 200),

0 commit comments

Comments
 (0)