PyTorch训练MNIST数据集3

不要绝望总会慢慢变强

于 2020-09-07 20:56:12 发布

阅读量392

点赞数 1

分类专栏： PyTorch

本文链接：https://blog.youkuaiyun.com/luoshiyong123/article/details/108455456

版权

PyTorch 专栏收录该内容

18 篇文章

订阅专栏

本文旨在探索各种不同的改变对于训练精确率的影响

1.增加全连接层

原来的全连接层由320->10：

self.f1 = torch.nn.Linear(320,10)

现在改为：每一个线性单元采用relu激活函数

        self.f1 = torch.nn.Linear(320,160)
        self.f2 = torch.nn.Linear(160,80)
        self.f3 = torch.nn.Linear(80,40)
        self.f4 = torch.nn.Linear(40,10)

结果：训练20次可达到99%