Shortcuts

mmengine.runner.load_state_dict

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

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.

参数:
  • 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.