DeepSpeed 工具

DeepSpeedPlugin

get_active_deepspeed_plugin

[[autodoc]] utils.get_active_deepspeed_plugin

[[autodoc]] utils.DeepSpeedPlugin

[[autodoc]] utils.deepspeed.DummyScheduler

DeepSpeedEngineWrapper

[[autodoc]] utils.deepspeed.DeepSpeedEngineWrapper

DeepSpeedOptimizerWrapper

[[autodoc]] utils.deepspeed.DeepSpeedOptimizerWrapper

DeepSpeedSchedulerWrapper

[[autodoc]] utils.deepspeed.DeepSpeedSchedulerWrapper

DummyOptim

[[autodoc]] utils.deepspeed.DummyOptim

DummyScheduler