mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
bc264269c9
* add an alternate RMSprop opt that applies eps like TF * add bn params for passing through alternates and changing defaults to TF style
4 lines
153 B
Python
4 lines
153 B
Python
from optim.adabound import AdaBound
|
|
from optim.nadam import Nadam
|
|
from optim.rmsprop_tf import RMSpropTF
|
|
from optim.optim_factory import create_optimizer |