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

11 lines
300 B
Python

from .nadam import Nadam
from .rmsprop_tf import RMSpropTF
from .adamw import AdamW
from .radam import RAdam
from .novograd import NovoGrad
from .nvnovograd import NvNovoGrad
from .lookahead import Lookahead
from .adamp import AdamP
from .sgdp import SGDP
from .optim_factory import create_optimizer