FDConv替换了Conv2d

[chenxing9527]

您好，我在yolov5中使用FDConv替换了Con2d，如图所示；

出现报错，尝试修改batchsize以及丢弃最后一批都无法解决，请问一下还有其他原因吗？
报错如下：
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 16, 1, 1])

[liu-5658]

self.bn = nn.BatchNorm2d(attention_channel)这里出问题了，换下就好了

[ElliottZhen]

你好，请问下具体是怎么操作?
@liu-5658 @chenxing9527

[ElliottZhen]

你好， batchsize为1就报错。ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 16, 1, 1])
有解决办法吗？
@Linwei-Chen

[Linwei-Chen]

这是nn.BatchNorm2d的问题，BatchNorm训练时需要用批量计算均值方差；将其换成其他的norm即可，例如groupnorm

[ElliottZhen]

太感谢您了！

[Linwei-Chen]

🥳