pytorch学习(三):mnist数据集测试和训练（二）卷积神经网络的构建与训练

1.构建卷积神经网络 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(1, 10, kernel_size=5) self.conv2 = nn.Conv2d(10, 20, kernel_size=5)

>>阅读原文<<