参考链接: 卷积神经网络在mnist数据集上的应用 Python
本文将为尽可能多的代码作注释,用PyTorch实现对手写数字数据集MNIST的分类,我也是一个PyTorch的初学者,如果你也是一个刚学...=nn.Sequential( #输入的数据集里的图像大小为28行*28列*1通道
nn.Conv2d(1,64,kernel_size=3,padding=1), #卷积后...,大小变为28*28*64
nn.BatchNorm2d(64), #卷积完之后来一下batch normalization,加快收敛
nn.ReLU...(), #加入ReLU作为激活函数
nn.Conv2d(64,128,kernel_size=3,padding=1), #卷积后张量大小变为28*28*128
...,download=True) #测试集
train_loader=DataLoader(train_dataset,batch_size=batch_size,shuffle=True) #数据迭代器