加入CAattention报错

[x2ss]

谢谢up主的分享，不知道up主有没有把CAattention加进去试过，我把CoorAttention加进去后，运行yolo.py里的
if opt.profile:
img = torch.rand(8 if torch.cuda.is_available() else 1, 3, 640, 640).to(device)
print('image size is', img.size())
y = model(img, profile=True)
会报错：
kernel = torch.DoubleTensor([*(x[0].shape[2:])]) // torch.DoubleTensor(list((m.output_size,))).squeeze()
TypeError: must be real number, not NoneType，是从x_h = self.pool_h(x) # h avg pool过来的。
如果up主试过，不知道是否遇到这个问题？遇到这个问题是如何解决的？
谢谢！！！

[x2ss]

我单独运行commn.py里的下面代码也没有问题，
if name == 'main':
x = torch.ones(1, 16, 2, 2)
a = CoorAttention(16, 16)
print(a(x).size())
就是放到yolov5里不行，不知道up主试过没有。
谢谢您的解答！

[x2ss]

应该是yolov5的什么问题，虽然这里报错，但可以训练