RuntimeError: Given input size: (256x1x1). Calculated output size: (256x0x0). Output size is too small
1.初始网络模型
原始网络模型对于cifar10使用了六层卷积,且每层卷积后接一层池化。 池化大小为(2*2,stride2)。
2.…
最近在跑一个联邦学习项目,用到的数据集是cifa-10,跑代码时出现以下报错:
RuntimeError: Given input size: (256x4x4). Calculated output size: (256x0x0). Output size is too small问题说的是,前向传播时,神经网络…