报错原因:torch.load在加载模型时实际需要参考网络结构构建出一个对应网络的实例对象,需要参考源class,所以报此错误原因是搭建的网络class 不在运行文件中
解决:
(1)简单解决:将网络class 代码复制一份到运行文件中
(2)参考我的博客:python动态导入自己写的模块或类_Wsyoneself的博客-CSDN博客
将网络所在文件的内容全部导入,如
from xxxx import *
再次测试即可正确运行~

报错原因:torch.load在加载模型时实际需要参考网络结构构建出一个对应网络的实例对象,需要参考源class,所以报此错误原因是搭建的网络class 不在运行文件中
解决:
(1)简单解决:将网络class 代码复制一份到运行文件中
(2)参考我的博客:python动态导入自己写的模块或类_Wsyoneself的博客-CSDN博客
将网络所在文件的内容全部导入,如
from xxxx import *
再次测试即可正确运行~