因为这个问题报错的时候,我一直以为是输入的数据有问题,而且在网上查找资料的时候,类似的问题全是loss function的问题,没有关于nn.Conv2d()相关的,所以记录一下,也让有类似问题的朋友思路更加明确,而不是在没有错误的地方解决'错误'
其实和loss function实例化相关的错误一样, 这个错误也是在实例化的时候出错的,和输入数据,卷积参数之类的没有关系
1、首先在网络模型外定义相关的类
可以看到需要初始化的参数为 input_channels
2、在网络初始化时定义引用该类并赋予初始值
3、forward()函数中进行调用(出现问题所在)
调用时一定要调用 self.attention(x) 而不是 self_ attention,当时问题就是出现在这里才报错了
希望能给大家在解决错误时一点思路!