|
在运行代码时遇到了这个错误，显示错误对应代码中的state_dict，找到对应的语句
修改前：net.load_state_dict(torch.load(model_para_path))
修改后：net.load_state_dict(torch.load(model_para_path),False)
我遇到的问题得到解决。经查阅资料，我理解的原因是state_dict的四个参数之一:_module 可以用来判断模型当前运行环境与之前是否相同，在默认情况下是True（纯属个人理解，勿喷）
参考：https://zhuanlan.zhihu.com/p/98563721
https://blog.csdn.net/qq_42052956/article/details/96423514
| 
转播
