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


  按照网上的方法在load_state_dict函数参数中加入strict=False,不顶用。

  最后发现是重装系统后,配置的pytorch环境和依赖包版本都和之前保持一致,之前训练的权重文件直接拿来test就出现这个问题。可能还是哪里环境发生变化了,只有重新训练模型得到新的权重文件,再test,问题解决

你可能感兴趣的:(深度学习,深度学习,python)