28 lines
831 B
Plaintext
28 lines
831 B
Plaintext
|
# Optimization
|
||
|
|
|
||
|
|
This page contains the API reference documentation for learning rate optimizers included in `timm`.
|
||
|
|
|
||
|
|
## Optimizers
|
||
|
|
|
||
|
|
### Factory functions
|
||
|
|
|
||
|
|
[[autodoc]] timm.optim.optim_factory.create_optimizer
|
||
|
|
[[autodoc]] timm.optim.optim_factory.create_optimizer_v2
|
||
|
|
|
||
|
|
### Optimizer Classes
|
||
|
|
|
||
|
|
[[autodoc]] timm.optim.adabelief.AdaBelief
|
||
|
|
[[autodoc]] timm.optim.adafactor.Adafactor
|
||
|
|
[[autodoc]] timm.optim.adahessian.Adahessian
|
||
|
|
[[autodoc]] timm.optim.adamp.AdamP
|
||
|
|
[[autodoc]] timm.optim.adamw.AdamW
|
||
|
|
[[autodoc]] timm.optim.lamb.Lamb
|
||
|
|
[[autodoc]] timm.optim.lars.Lars
|
||
|
|
[[autodoc]] timm.optim.lookahead.Lookahead
|
||
|
|
[[autodoc]] timm.optim.madgrad.MADGRAD
|
||
|
|
[[autodoc]] timm.optim.nadam.Nadam
|
||
|
|
[[autodoc]] timm.optim.nvnovograd.NvNovoGrad
|
||
|
|
[[autodoc]] timm.optim.radam.RAdam
|
||
|
|
[[autodoc]] timm.optim.rmsprop_tf.RMSpropTF
|
||
|
|
[[autodoc]] timm.optim.sgdp.SGDP
|