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

9 lines
252 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 .optim_factory import create_optimizer