pytorch-image-models/optim/__init__.py

from optim.nadam import Nadam
from optim.rmsprop_tf import RMSpropTF
from optim.optim_factory import create_optimizer
Uniform pretrained model handling. * All models have 'default_cfgs' dict * load/resume/pretrained helpers factored out * pretrained load operates on state_dict based on default_cfg * test all models in validate * schedule, optim factor factored out * test time pool wrapper applied based on default_cfg 2019-04-12 12:32:16 +08:00			`from optim.nadam import Nadam`
Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related * add an alternate RMSprop opt that applies eps like TF * add bn params for passing through alternates and changing defaults to TF style 2019-04-22 06:54:28 +08:00			`from optim.rmsprop_tf import RMSpropTF`
Remove adabound optimizer, never got it working well on larger datasets 2019-05-15 09:29:34 +08:00			`from optim.optim_factory import create_optimizer`