pytorch-image-models/timm/optim/__init__.py

20 lines
600 B
Python
Raw Normal View History

from .adabelief import AdaBelief
from .adafactor import Adafactor
2024-11-09 01:10:17 +08:00
from .adafactor_bv import AdafactorBigVision
from .adahessian import Adahessian
from .adamp import AdamP
from .adamw import AdamW
2022-12-15 11:37:29 +08:00
from .adan import Adan
2024-11-09 01:10:17 +08:00
from .adopt import Adopt
from .lamb import Lamb
from .lars import Lars
2024-11-09 01:10:17 +08:00
from .lion import Lion
from .lookahead import Lookahead
from .madgrad import MADGRAD
from .nadam import Nadam
from .nvnovograd import NvNovoGrad
from .radam import RAdam
from .rmsprop_tf import RMSpropTF
from .sgdp import SGDP
2024-11-09 01:10:17 +08:00
from .optim_factory import create_optimizer, create_optimizer_v2, optimizer_kwargs