每天五分钟深度学习PyTorch：搭建卷积神经网络完成手写字体识别

本文重点

前面的一篇文章我们学习了使用pytorch搭建卷积神经网络，本文我们使用刚刚搭建的卷积神经网络完成手写字体识别的任务

手写字体识别是之前进行介绍过的常用的数据集了，然后卷积神经网络如何搭建前面已经进行介绍了，下面我们将再详细的介绍一下如何使用卷积神经网络完成手写字体识别。

算法模型

class CNN(nn.Module):

def __init__(self):

super(CNN,self).__init__()

self.layer1=nn.Sequential( nn.Conv2d(1,16,kernel_size=3), nn.BatchNorm2d(16), nn.ReLU(inplace=True), ) self.layer2=nn.Sequential( nn.Conv2d(16,32,kernel_size=3), nn.BatchNorm2d(32), nn.ReLU(inplace=True), nn.MaxPool2d(kernel_size=2,stride=2) )

self.layer3=nn.Sequential( nn.Conv2d(32,64,kernel_size=3), nn.BatchNorm2d(64), nn.ReLU(inplace=True) ) self.layer4=nn.Sequential( nn.Conv2d(64,128,kerne