[已解决]关于通道数的修改

季夏。。。

feat_img = net.FeatCNN(img.to(torch.float32).to(device=torch.device("cuda" if torch.cuda.is_available() else "cpu")))
            feat_arr = net.FeatRNN(arr.to(torch.float32).to(device=torch.device("cuda" if torch.cuda.is_available() else "cpu")))
            feat = torch.cat((feat_img, feat_arr), axis=1)

运行到这一部分的时候就显示：RuntimeError: Given groups=1, weight of size [32, 5, 3, 3], expected input[144, 3, 32, 32] to have 5 channels, but got 3 channels instead
不知道是什么原因，有没有大佬知道能帮忙看看的

最佳答案

月排行榜 / 总排行榜

suchocolate

2022-12-3 11:31:10

没研究神经网络，从报错上看，给的通道数不够，期望5个，只提供了3个。
最好把报错代码发全

跳转到最佳答案楼层

suchocolate

没研究神经网络，从报错上看，给的通道数不够，期望5个，只提供了3个。
最好把报错代码发全

ZhKQYu

贴一个训练的代码看看，应该是训练的时候，图片的通道数整错了