加载模型时发生错误RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX
解决方法:
model.load_state_dict(torch.load('models/params.pt'),strict=False)
在参数中加入strict=False
此类错误一部分原因是pytorch环境不同导致的
加载模型时发生错误RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX
解决方法:
model.load_state_dict(torch.load('models/params.pt'),strict=False)
在参数中加入strict=False
此类错误一部分原因是pytorch环境不同导致的