Skip to content

Commit 75435da

Browse files
committed
update: recipes
1 parent 859741d commit 75435da

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/constants.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
MSVAG,
88
OPTIMIZERS,
99
PNM,
10+
QHM,
1011
SGDP,
1112
SGDW,
1213
SM3,
@@ -343,6 +344,8 @@
343344
(AggMo, {'lr': 5e0, 'weight_decay': 1e-3, 'weight_decouple': True}, 5),
344345
(QHAdam, {'lr': 1e0, 'nus': (0.9, 0.9), 'weight_decay': 1e-3}, 5),
345346
(QHAdam, {'lr': 1e0, 'weight_decay': 1e-3, 'weight_decouple': True}, 5),
347+
(QHM, {'lr': 1e0, 'weight_decay': 1e-3}, 5),
348+
(QHM, {'lr': 1e0, 'weight_decay': 1e-3, 'weight_decouple': True}, 5),
346349
]
347350
ADANORM_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [
348351
(AdaBelief, {'lr': 5e-1, 'weight_decay': 1e-3, 'adanorm': True}, 10),

0 commit comments

Comments
 (0)