Skip to content

Commit 93c5dd9

Browse files
committed
update: recipes
1 parent c4548cf commit 93c5dd9

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
@@ -6,6 +6,7 @@
66
MADGRAD,
77
MSVAG,
88
OPTIMIZERS,
9+
PID,
910
PNM,
1011
QHM,
1112
SGDP,
@@ -346,6 +347,8 @@
346347
(QHAdam, {'lr': 1e0, 'weight_decay': 1e-3, 'weight_decouple': True}, 5),
347348
(QHM, {'lr': 1e0, 'weight_decay': 1e-3}, 5),
348349
(QHM, {'lr': 1e0, 'weight_decay': 1e-3, 'weight_decouple': True}, 5),
350+
(PID, {'lr': 1e0, 'momentum': 0.9, 'dampening': 1.0, 'weight_decay': 1e-3}, 5),
351+
(PID, {'lr': 1e0, 'momentum': 0.9, 'dampening': 1.0, 'weight_decay': 1e-3, 'weight_decouple': True}, 5),
349352
]
350353
ADANORM_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [
351354
(AdaBelief, {'lr': 5e-1, 'weight_decay': 1e-3, 'adanorm': True}, 10),

0 commit comments

Comments
 (0)