pytorch加载模型时报错 AttributeError: Can‘t get attribute ‘CNN‘ on ＜module ‘__main__‘＞

1、问题描述

在Pytorch加载模型时报错：
AttributeError: Can’t get attribute ‘CNN’ on <module ‘__ main __’ from 'E:\…

看了很多篇其他人的分享，都比较复杂或者不适合我这种情况。

2、解决办法

将你之前训练模型时，定义模型的类全部复制到这个py文件中即可

对于我这种情况，即包括__init__方法和forward方法的整个CNN网络，如下：