Skip to content

Commit b68a3c1

Browse files
committed
update: Adan recipes
1 parent 46826b4 commit b68a3c1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/test_optimizers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,8 @@
8080
(AdaPNM, {'lr': 3e-1, 'weight_decay': 1e-3, 'amsgrad': False}, 500),
8181
(Nero, {'lr': 5e-1}, 200),
8282
(Nero, {'lr': 5e-1, 'constraints': False}, 200),
83-
(Adan, {'lr': 1e-1}, 200),
83+
(Adan, {'lr': 2e-1}, 200),
84+
(Adan, {'lr': 2e-1, 'weight_decay': 1e-4, 'use_gc': True}, 500),
8485
]
8586

8687
ADAMD_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [

0 commit comments

Comments
 (0)