pytorch-image-models/optim/__init__.py

4 lines
118 B
Python

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