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. Default: False.
logger (logging.Logger, optional) – Logger to log the error message. If not specified, print function will be used.