Given groups=1, weight of size 128 128 3 3, expected input[32, 256, 7, 7] to have 128 channels, but

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

错误描述：应该有128个通道，但是却得到256个通道

我做的是通过vgg16网络识别mnist数据集。
输入的图片是28 X 28 X 1
经过卷积池化后，进入全连接层的输入应当是1 X 1X 512
以上是我的计算。然后报错

解决方案：
发现是定义网络的时候，上下通道没有对应，导致错误！