RuntimeError: Boolean value of Tensor with more than one value is ambiguous

在debug突然出现这个错误,错误出现在卷积这一行。

self.conv = nn.Sequential(nn.Conv2d(in_channels=in_channel*2,out_channels=channel*2,kernel_size=3,stride=1,padding=1,groups=in_channel*2),
                                  nn.BatchNorm2d(in_channel*2),
                                  nn.ReLU(inplace=True))

然后仔细检查了卷积的定义,发现没有问题,然后网上找的答案都没有解决。
然后整体检查了代码终于发现了错误。在实例化的地方,我直接用y实例化了类,因此x就相当于输入通道,map相当于channel。所以在使用通道的卷积处发生了错误。
RuntimeError: Boolean value of Tensor with more than one value is ambiguous_第1张图片
将CA改为实例化的f就可以了。结果正常运行出来了,这是粗心导致的bug,害我花了好几分钟。RuntimeError: Boolean value of Tensor with more than one value is ambiguous_第2张图片

你可能感兴趣的:(pytorch踩坑,深度学习,python,计算机视觉)