Skip to content

Commit 7811dd2

Browse files
committed
update: CAME optimizer
1 parent bc93c1b commit 7811dd2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/constants.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -428,8 +428,8 @@
428428
(Prodigy, {'lr': 1e0, 'beta3': 0.999, 'weight_decay': 1e-3, 'safeguard_warmup': True}, 15),
429429
(PAdam, {'lr': 1e0, 'weight_decay': 1e-3}, 5),
430430
(Tiger, {'lr': 1e0, 'weight_decay': 1e-3}, 5),
431-
(CAME, {'lr': 7.5e-1, 'weight_decay': 1e-3}, 75),
432-
(CAME, {'lr': 7.5e-1, 'weight_decay': 1e-3, 'ams_bound': True}, 75),
431+
(CAME, {'lr': 7.5e-1, 'weight_decay': 1e-3}, 70),
432+
(CAME, {'lr': 7.5e-1, 'weight_decay': 1e-3, 'ams_bound': True}, 70),
433433
(Aida, {'lr': 1e0, 'weight_decay': 1e-3, 'ams_bound': True}, 5),
434434
(
435435
GaLore,

0 commit comments

Comments
 (0)