Simple CNN Model with pytorch and MNIST dataset is used. This code was for my pytorch practice so more optimization is needed for actual usage. You need to make a directory './data' and download MNIST dataset in the directory.
To train and evaluate the model:
$ python cnn.py
3 epochs * 1875 steps iteration result
cnn.py --> training loss: 0.110 / test accuracy: 95.12
- Insert Batch Normalization Layers
Sooyoung Moon / @symoon94