Skip to content

Commit e0acc05

Browse files
committed
update: recipes
1 parent ae8ea0f commit e0acc05

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/constants.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
AdaFactor,
2424
AdaHessian,
2525
Adai,
26+
Adalite,
2627
AdaMax,
2728
AdaMod,
2829
AdamP,
@@ -120,6 +121,8 @@
120121
'padam',
121122
'came',
122123
'aida',
124+
'galore',
125+
'adalite',
123126
]
124127

125128
VALID_LR_SCHEDULER_NAMES: List[str] = [
@@ -434,6 +437,7 @@
434437
{'lr': 5e-1, 'weight_decay': 1e-3, 'rank': 2, 'scale': 1.0, 'update_proj_gap': 2, 'projection_type': 'full'},
435438
5,
436439
),
440+
(Adalite, {'lr': 1e0, 'weight_decay': 1e-3}, 5),
437441
]
438442
ADANORM_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [
439443
(AdaBelief, {'lr': 5e-1, 'weight_decay': 1e-3, 'adanorm': True}, 10),

0 commit comments

Comments
 (0)