From 942726db317735bce319cbf3d5140fc2a1ffebc8 Mon Sep 17 00:00:00 2001 From: "alec.tu" Date: Thu, 27 Jul 2023 09:26:57 +0800 Subject: [PATCH] import lion in __init__.py --- timm/optim/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/timm/optim/__init__.py b/timm/optim/__init__.py index 0775638e..d2eb67ba 100644 --- a/timm/optim/__init__.py +++ b/timm/optim/__init__.py @@ -13,4 +13,5 @@ from .nvnovograd import NvNovoGrad from .radam import RAdam from .rmsprop_tf import RMSpropTF from .sgdp import SGDP +from .lion import Lion from .optim_factory import create_optimizer, create_optimizer_v2, optimizer_kwargs