Shortcuts

mmengine.runner.load_state_dict

mmengine.runner.load_state_dict(module, state_dict, strict=False, logger=None)[source]

Load state_dict to a module.

This method is modified from torch.nn.Module.load_state_dict(). Default value for strict is set to False and the message for param mismatch will be shown even if strict is False.

Parameters
  • module (Module) – Module that receives the state_dict.

  • state_dict (OrderedDict) – Weights.

  • strict (bool) – whether to strictly enforce that the keys in state_dict match the keys returned by this module’s state_dict() function. Defaults to False.

  • logger (logging.Logger, optional) – Logger to log the error message. If not specified, print function will be used.

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.