RuntimeError: Given groups=1, weight of size 64 4 4 4, expected input[1, 5, 256, 256] to have 4 chan

报错信息:RuntimeError: Given groups=1, weight of size 64 4 4 4, expected input[1, 5, 256, 256] to have 4 channels, but got 5 channels instead

在这里插入图片描述
大意为预期的输入图像为4通道,而得到的是5通道的图像从而导致报错

[1, 5, 256, 256] — 1代表batch_size 5代表通道数

解决方法:使用debug找到通道不匹配的地方进行修改

你可能感兴趣的:(#,Python)