pytorch复现文章的一个小报错

pytorch 文章复现的小报错

复现了一个backbone，第一次卷积就报错说我输入的图像通道有问题。具体见下图

说我的输入是16 * 32 * 510 * 510.

可以我的batch确实是16，但是我的图像是3 * 512 * 512，怎么可能是32 * 510 * 510呢。

于是我打印了一下transform之后的inputs的shape

没有发现任何问题。

报错位置代码

就是第一层卷积，扩一下通道

肿么了

输入[16, 3, 512, 512]，经过第一层卷积后变成[16, 32, 510, 510]，这个错误应该说的是后面，看看你第二层卷积的输入通道是不是32

那你可以尝试把属性带上，在conv2d括号里按ctrl+p就会有提示

复现了下你的错误，说明你这一层卷积输入的形状是(16,32,510,510)，检查下这一层之前有什么其它操作。