CNN-for-MNIST

使用CNN实现对手写数字的识别

Requirement

Using the PyTorch framework for neural networks
*PyTorch version 1.2.0
*Python version 3.7.3

Usage

train.py为对数据集的单次EPOCH训练
train_iter.py为对数据集的多次EPOCH训练

如果觉得不错！点个star！