RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX

加载模型时发生错误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环境不同导致的

你可能感兴趣的:(Bug,pytorch,深度学习,神经网络)