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 forstrict
is set toFalse
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’sstate_dict()
function. Defaults to False.logger (
logging.Logger
, optional) – Logger to log the error message. If not specified, print function will be used.