mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
e35ed5fd2e
* add ApexOptimWrapper * typo fix * add apex amp.initialize in optim_context * assert apex_amp * polish code * add parameters of apex_amp.initialize * add docs * polish code * polish code * polish code * fix calling of apex amp load_state_dict * polish * add comments * Update apex_optimizer_wrapper.py * Update apex_optimizer_wrapper.py --------- Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
70 lines
1.2 KiB
ReStructuredText
70 lines
1.2 KiB
ReStructuredText
.. role:: hidden
|
|
:class: hidden-section
|
|
|
|
mmengine.optim
|
|
===================================
|
|
|
|
.. contents:: mmengine.optim
|
|
:depth: 2
|
|
:local:
|
|
:backlinks: top
|
|
|
|
.. currentmodule:: mmengine.optim
|
|
|
|
Optimizer
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
AmpOptimWrapper
|
|
ApexOptimWrapper
|
|
OptimWrapper
|
|
OptimWrapperDict
|
|
DefaultOptimWrapperConstructor
|
|
ZeroRedundancyOptimizer
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
build_optim_wrapper
|
|
|
|
Scheduler
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
_ParamScheduler
|
|
ConstantLR
|
|
ConstantMomentum
|
|
ConstantParamScheduler
|
|
CosineAnnealingLR
|
|
CosineAnnealingMomentum
|
|
CosineAnnealingParamScheduler
|
|
ExponentialLR
|
|
ExponentialMomentum
|
|
ExponentialParamScheduler
|
|
LinearLR
|
|
LinearMomentum
|
|
LinearParamScheduler
|
|
MultiStepLR
|
|
MultiStepMomentum
|
|
MultiStepParamScheduler
|
|
OneCycleLR
|
|
OneCycleParamScheduler
|
|
PolyLR
|
|
PolyMomentum
|
|
PolyParamScheduler
|
|
StepLR
|
|
StepMomentum
|
|
StepParamScheduler
|
|
ReduceOnPlateauLR
|
|
ReduceOnPlateauMomentum
|
|
ReduceOnPlateauParamScheduler
|