2020
10-08
10-08
Pytorch框架实现mnist手写库识别(与tensorflow对比)
前言最近在学习过程中需要用到pytorch框架,简单学习了一下,写了一个简单的案例,记录一下pytorch中搭建一个识别网络基础的东西。对应一位博主写的tensorflow的识别mnist数据集,将其改为pytorch框架,也可以详细看到两个框架大体的区别。Tensorflow版本转载来源(CSDN博主「兔八哥1024」):https://www.jb51.net/article/191157.htmPytorch实战mnist手写数字识别#需要导入的包importtorchimporttorch.nnasnn#用于构建网络层...
继续阅读 >