pytorch手写数字识别

项目介绍

本项目使用pytorch实现了简单的手写数字识别，运用了三种方法

1.两层全连接网络

2.四层全连接网络

3.四层卷积神经网络

通过在代码中修改相应的参数，即可选择对应的方法，参考注释

文件介绍

displayMNIST.py是显示手写数字识别（NMIST）数据集

mnistCPU.py是通过CPU进行训练

mnistGPU.py是通过GPU进行训练

运行方式

先直接运行mnistCPU.py或者mnistGPU.py就可以下载数据集，数据集下载成功后，就开始了训练，如果要查看数据集，对数据集进行显示，可以运行displayMNIST.py代码

环境要求

python3，pytorch