pytorch 文章复现的小报错
复现了一个backbone,第一次卷积就报错说我输入的图像通道有问题。具体见下图
说我的输入是16 * 32 * 510 * 510.
可以我的batch确实是16,但是我的图像是3 * 512 * 512,怎么可能是32 * 510 * 510呢。
于是我打印了一下transform之后的inputs的shape
没有发现任何问题。
报错位置代码
就是第一层卷积,扩一下通道
肿么了
说我的输入是16 * 32 * 510 * 510.
可以我的batch确实是16,但是我的图像是3 * 512 * 512,怎么可能是32 * 510 * 510呢。
于是我打印了一下transform之后的inputs的shape
没有发现任何问题。
报错位置代码
就是第一层卷积,扩一下通道
肿么了
输入[16, 3, 512, 512],经过第一层卷积后变成[16, 32, 510, 510],这个错误应该说的是后面,看看你第二层卷积的输入通道是不是32