DeepSpeedOptimWrapper¶
- class mmengine._strategy.deepspeed.DeepSpeedOptimWrapper(optimizer)[源代码]¶
-
- load_state_dict(state_dict)[源代码]¶
A wrapper of
Optimizer.load_state_dict
. load the state dict ofoptimizer
.Provide unified
load_state_dict
interface compatible with automatic mixed precision training. Subclass can overload this method to implement the required logic. For example, the state dictionary of GradScaler should be loaded when training withtorch.cuda.amp
.- 参数:
state_dict (dict) – The state dictionary of
optimizer
.- 返回类型:
None