pytorch-image-models/optim/__init__.py

3 lines
114 B
Python

from optim.adabound import AdaBound
from optim.nadam import Nadam
from optim.optim_factory import create_optimizer