From f096a438c087b3334d466d84b0484893e4b0448c Mon Sep 17 00:00:00 2001 From: Sangdoo Yun Date: Sat, 25 Jul 2020 12:30:34 +0900 Subject: [PATCH] Update __init__.py copy files of adamp & sgdp --- timm/optim/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/timm/optim/__init__.py b/timm/optim/__init__.py index 50906e9c..3727f0be 100644 --- 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