Skip to content

Commit b087e56

Browse files
committed
update: RaLAMB-degenerated_to_sgd configuration
1 parent 8dc56ff commit b087e56

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_optimizers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@
6060
(LARS, {'lr': 1e-1, 'weight_decay': 1e-3}, 500),
6161
(RaLamb, {'lr': 2e-1, 'weight_decay': 1e-3}, 500),
6262
(RaLamb, {'lr': 5e-1, 'weight_decay': 1e-3, 'pre_norm': True}, 500),
63-
# (RaLamb, {'lr': 1e-1, 'weight_decay': 1e-3, 'degenerated_to_sgd': True}, 200),
63+
(RaLamb, {'lr': 1e-1, 'weight_decay': 1e-3, 'degenerated_to_sgd': True}, 200),
6464
(MADGRAD, {'lr': 1e-2, 'weight_decay': 1e-3}, 500),
6565
(MADGRAD, {'lr': 1e-2, 'weight_decay': 1e-3, 'eps': 0.0}, 500),
6666
(MADGRAD, {'lr': 1e-2, 'weight_decay': 1e-3, 'momentum': 0.0}, 500),

0 commit comments

Comments
 (0)