RuntimeError: Error(s) in loading state_dict for ResNet:

对于这个问题,我的问题需要修改的是

修改前:

net.load_state_dict(torch.load(path_net))

修改后:

net.load_state_dict(torch.load(path_net),False)

具体原因见函数声明:

    def load_state_dict(self, state_dict, strict=True):
        r"""Copies parameters and buffers from :attr:`state_dict` into
        this module and its descendants. If :attr:`strict` is ``True``, then
        the keys of :attr:`state_dict` must exactly match the keys returned
        by this module's :meth:`~torch.nn.Module.state_dict` function.

        Arguments:
            state_dict (dict): a dict containing parameters and
                persistent buffers.
            strict (bool, optional): whether to strictly enforce that the keys
                in :attr:`state_dict` match the keys returned by this module's
                :meth:`~torch.nn.Module.state_dict` function. Default: ``True``

 

你可能感兴趣的:(Error)