OSVAI/ODConv

调用summary是报错,是什么原因呢?

dddlli opened this issue · 2 comments

ODConv/modules/odconv.py

Lines 56 to 58 in 60a4691

@staticmethod
def skip(_):
return 1.0

ODConv/modules/odconv.py

Lines 78 to 83 in 60a4691

def forward(self, x):
x = self.avgpool(x)
x = self.fc(x)
x = self.bn(x)
x = self.relu(x)
return self.func_channel(x), self.func_filter(x), self.func_spatial(x), self.func_kernel(x)

该错误与ODConv2d的Attention实现方式有关,在kernel size为1/kernel number为1/depth-wise卷积等不同情况下,channel/filter/spatial/kernel attention中的某些attention机制将不被应用(例如ODConv不会对1×1卷积层使用spatial attention),我们的代码实现中会将相应的attention置为浮点数1.0。 针对此问题,可以先判断查看size的对象是否为tensor类型。

你好,我也遇到了同样的问题,查看size的对象是LongTensor类型,不知道要怎么解决T_T