2023-06-27 11:32:39 +08:00
|
|
|
.. role:: hidden
|
|
|
|
|
:class: hidden-section
|
|
|
|
|
|
|
|
|
|
mmengine._strategy
|
|
|
|
|
===================================
|
|
|
|
|
|
|
|
|
|
.. currentmodule:: mmengine._strategy
|
|
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: generated
|
|
|
|
|
:nosignatures:
|
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
|
|
BaseStrategy
|
|
|
|
|
SingleDeviceStrategy
|
|
|
|
|
DDPStrategy
|
|
|
|
|
DeepSpeedStrategy
|
2023-06-28 16:50:52 +08:00
|
|
|
FSDPStrategy
|
2023-08-18 15:09:35 +08:00
|
|
|
ColossalAIStrategy
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.. currentmodule:: mmengine._strategy.deepspeed
|
|
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: generated
|
|
|
|
|
:nosignatures:
|
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
|
|
MMDeepSpeedEngineWrapper
|
|
|
|
|
DeepSpeedOptimWrapper
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.. currentmodule:: mmengine._strategy.colossalai
|
|
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: generated
|
|
|
|
|
:nosignatures:
|
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
|
|
CollosalAIModelWrapper
|
2023-08-22 16:13:55 +08:00
|
|
|
ColossalAIOptimWrapper
|
