Update __init__.py

Add `adamp` optimizer.
5 years ago · b24fb86fea
parent 17f4dd2778
commit b24fb86fea
1 changed files with 1 additions and 0 deletions
--- a/timm/optim/init.py
+++ b/timm/optim/init.py
@ -5,4 +5,5 @@ from .radam import RAdam
 from .novograd import NovoGrad
 from .nvnovograd import NvNovoGrad
 from .lookahead import Lookahead
+from adamp import AdamP
 from .optim_factory import create_optimizer