Back to Pytorch Image Models

Optimization

hfdocs/source/reference/optimizers.mdx

1.0.261.0 KB
Original Source

Optimization

This page contains the API reference documentation for learning rate optimizers included in timm.

Optimizers

Factory functions

[[autodoc]] timm.optim.create_optimizer_v2 [[autodoc]] timm.optim.list_optimizers [[autodoc]] timm.optim.get_optimizer_class

Optimizer Classes

[[autodoc]] timm.optim.adabelief.AdaBelief [[autodoc]] timm.optim.adafactor.Adafactor [[autodoc]] timm.optim.adafactor_bv.AdafactorBigVision [[autodoc]] timm.optim.adahessian.Adahessian [[autodoc]] timm.optim.adamp.AdamP [[autodoc]] timm.optim.adan.Adan [[autodoc]] timm.optim.adopt.Adopt [[autodoc]] timm.optim.lamb.Lamb [[autodoc]] timm.optim.laprop.LaProp [[autodoc]] timm.optim.lars.Lars [[autodoc]] timm.optim.lion.Lion [[autodoc]] timm.optim.lookahead.Lookahead [[autodoc]] timm.optim.madgrad.MADGRAD [[autodoc]] timm.optim.mars.Mars [[autodoc]] timm.optim.nadamw.NAdamW [[autodoc]] timm.optim.nvnovograd.NvNovoGrad [[autodoc]] timm.optim.rmsprop_tf.RMSpropTF [[autodoc]] timm.optim.sgdp.SGDP [[autodoc]] timm.optim.sgdw.SGDW