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