Update __init__.py

copy files of adamp & sgdp
5 years ago · f096a438c0
parent 4fae5f4c0e
commit f096a438c0
1 changed files with 2 additions and 1 deletions
--- a/timm/optim/init.py
+++ b/timm/optim/init.py
@ -5,5 +5,6 @@ from .radam import RAdam
 from .novograd import NovoGrad
 from .nvnovograd import NvNovoGrad
 from .lookahead import Lookahead
-from adamp import AdamP
+from .adamp import AdamP
+from .adamp import SGDP
 from .optim_factory import create_optimizer