解决Given groups=1,weight [128,64,4,4], so expected inputs [32,32,64,64] to have 64 channels类似错误

在测试网络的时候报出了这样的错误：

RuntimeError: Given groups=1, weight of size [128, 128, 3, 3], expected input[64, 64, 32, 32] to have 128 channels, but got 64 channels instead

经过检查发现网络出现了错误：

解决办法：

此外还有可能是卷积计算出错，导致input和output对应不起来，那就需要自行计算了。