Skip to content

Commit 2b16874

Browse files
committed
update: test cases
1 parent 830f7e1 commit 2b16874

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

tests/constants.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@
7171
RAdam,
7272
Ranger,
7373
Ranger21,
74+
Ranger25,
7475
ScalableShampoo,
7576
ScheduleFreeAdamW,
7677
ScheduleFreeRAdam,
@@ -543,6 +544,7 @@
543544
(SGDSaI, {'lr': 1e0}, 15),
544545
(SGDSaI, {'lr': 1e0, 'momentum': 0.0}, 15),
545546
(Grams, {'lr': 1e-1, 'weight_decay': 1e-3}, 5),
547+
(Ranger25, {'lr': 1e-1}, 25),
546548
]
547549
ADANORM_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [
548550
(AdaBelief, {'lr': 5e-1, 'weight_decay': 1e-3, 'adanorm': True}, 10),

tests/test_load_modules.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,9 @@ def test_load_lr_scheduler_invalid(invalid_lr_scheduler_names):
3434

3535

3636
def test_get_supported_optimizers():
37-
assert len(get_supported_optimizers()) == 86
37+
assert len(get_supported_optimizers()) == 87
3838
assert len(get_supported_optimizers('adam*')) == 7
39-
assert len(get_supported_optimizers(['adam*', 'ranger*'])) == 9
39+
assert len(get_supported_optimizers(['adam*', 'ranger*'])) == 10
4040

4141

4242
def test_get_supported_lr_schedulers():

0 commit comments

Comments
 (0)