mnist_train.py

立刻就能跑的pytorch代码，用神经网络训练MNIST手写数据集，有可视化查看效果的plot模块。三层激活网络，包含前向传递等模块，输出图像识别率。