diff --git a/timm/optim/__init__.py b/timm/optim/__init__.py index 7ee4958e..0775638e 100644 --- a/timm/optim/__init__.py +++ b/timm/optim/__init__.py @@ -3,6 +3,7 @@ from .adafactor import Adafactor from .adahessian import Adahessian from .adamp import AdamP from .adamw import AdamW +from .adan import Adan from .lamb import Lamb from .lars import Lars from .lookahead import Lookahead