每天五分钟深度学习PyTorch：基于全连接神经网络完成手写字体识别

本文重点

上一节我们学习了搭建普通的全连接神经网络，我们现在用它来解决一个实际问题，我们用它跑一下手写字体识别的数据，然后看看它的效果如何。

网络模型

class ThreeNet(nn.Module) :

def __init__ (self,in_dim,n_hidden_1,n_hidden_2,out_dim):

super(ThreeNet, self).__init__()

self.layer1 = nn.Linear (in_dim, n_hidden_1)

self.layer2 = nn.Linear(n_hidden_1,n_hidden_2)

self.layer3 = nn.Linear(n_hidden_2, out_dim)

def forward(self,x):

x =self.layer1(x)

x =self.layer2(x)

x =self.layer3(x)

return x