CV小白实践--实现MNIST手写数字识别时遇到的问题
1.RuntimeError: size mismatch, m1: [800 x 4], m2: [320 x 50]问题原因:这个问题出现在神经网络最后一层卷积层与第一层全连接层之间。首先来看一下我实现的神经网络的结构。 def __init__(self): super(Net,self).__init__() self.conv1 = nn.Conv2d(1,10,5) self.pool = nn.MaxPool2d(kernel_size
原创
2020-10-09 15:53:39 ·
535 阅读 ·
0 评论