Shortcuts

mmengine.runner.get_state_dict

mmengine.runner.get_state_dict(module, destination=None, prefix='', keep_vars=False)[source]

Returns a dictionary containing a whole state of the module.

Both parameters and persistent buffers (e.g. running averages) are included. Keys are corresponding parameter and buffer names. This method is modified from torch.nn.Module.state_dict() to recursively check parallel module in case that the model has a complicated structure, e.g., nn.Module(nn.Module(DDP)).

Parameters
  • module (nn.Module) – The module to generate state_dict.

  • destination (OrderedDict) – Returned dict for the state of the module.

  • prefix (str) – Prefix of the key.

  • keep_vars (bool) – Whether to keep the variable property of the parameters. Defaults to False.

Returns

A dictionary containing a whole state of the module.

Return type

dict

Read the Docs v: v0.8.3
Versions
latest
stable
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
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.