From b24fb86feaba34efab4122e8ccf3df99c2882939 Mon Sep 17 00:00:00 2001 From: Sangdoo Yun Date: Wed, 22 Jul 2020 11:02:05 +0900 Subject: [PATCH] Update __init__.py Add `adamp` optimizer. --- timm/optim/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/timm/optim/__init__.py b/timm/optim/__init__.py index 994b36d2..50906e9c 100644 --- 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