Skip to content

Commit c4548cf

Browse files
committed
update: PID optimizer
1 parent c1879d9 commit c4548cf

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pytorch_optimizer/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
from pytorch_optimizer.optimizer.nero import Nero
4747
from pytorch_optimizer.optimizer.novograd import NovoGrad
4848
from pytorch_optimizer.optimizer.pcgrad import PCGrad
49+
from pytorch_optimizer.optimizer.pid import PID
4950
from pytorch_optimizer.optimizer.pnm import PNM
5051
from pytorch_optimizer.optimizer.qhadam import QHAdam
5152
from pytorch_optimizer.optimizer.qhm import QHM
@@ -89,6 +90,7 @@
8990
OPTIMIZER_LIST: List[OPTIMIZER] = [
9091
AdaBelief,
9192
AdaBound,
93+
PID,
9294
AdamP,
9395
Adai,
9496
Adan,

0 commit comments

Comments
 (0)