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

4 lines
103 B
Python

from .nadam import Nadam
from .rmsprop_tf import RMSpropTF
from .optim_factory import create_optimizer