每天五分钟深度学习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
原文地址:https://blog.csdn.net/huanfeng_AI/article/details/143651867
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.kler.cn/a/387388.html 如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.kler.cn/a/387388.html 如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!