mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
fac58f609a
* Add some of the trendy new optimizers. Decent results but not clearly better than the standards. * Can create a None scheduler for constant LR * ResNet defaults to zero_init of last BN in residual * add resnet50d config
8 lines
217 B
Python
8 lines
217 B
Python
from .nadam import Nadam
|
|
from .rmsprop_tf import RMSpropTF
|
|
from .adamw import AdamW
|
|
from .radam import RAdam
|
|
from .novograd import NovoGrad
|
|
from .lookahead import Lookahead
|
|
from .optim_factory import create_optimizer
|