mmengine.optim.build_optim_wrapper¶
- mmengine.optim.build_optim_wrapper(model, cfg)[源代码]¶
Build function of OptimWrapper.
If
constructor
is set in thecfg
, this method will build an optimizer wrapper constructor, and use optimizer wrapper constructor to build the optimizer wrapper. Ifconstructor
is not set, theDefaultOptimWrapperConstructor
will be used by default.- 参数:
model (nn.Module) – Model to be optimized.
cfg (dict) – Config of optimizer wrapper, optimizer constructor and optimizer.
- 返回:
The built optimizer wrapper.
- 返回类型: