Shortcuts

mmengine.optim.build_optim_wrapper

mmengine.optim.build_optim_wrapper(model, cfg)[source]

Build function of OptimWrapper.

If constructor is set in the cfg, this method will build an optimizer wrapper constructor, and use optimizer wrapper constructor to build the optimizer wrapper. If constructor is not set, the DefaultOptimWrapperConstructor will be used by default.

Parameters:
  • model (nn.Module) – Model to be optimized.

  • cfg (dict) – Config of optimizer wrapper, optimizer constructor and optimizer.

Returns:

The built optimizer wrapper.

Return type:

OptimWrapper

Read the Docs v: latest
Versions
latest
stable
v0.10.3
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.