mnist_pytorch

基于Pytorch框架设计实现MNIST手写数字识别

python : 3.6

torch : 1.4.0

torchvision : 0.5.0

model.pkl保存的模型文件大小为24.5 MB，可以一起下载直接跑测试，亦可以单个main.py直接运行，训练+测试。

mnist数据过大故不上传到GitHub了，直接运行代码即可自动下载到./data/文件中。