Skip to content

symoon94/CNN-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 

Repository files navigation

CNN-pytorch

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.

Usage

To train and evaluate the model:

$ python cnn.py

Performance

3 epochs * 1875 steps iteration result

cnn.py --> training loss: 0.110 / test accuracy: 95.12

To-do

  • Insert Batch Normalization Layers

Author

Sooyoung Moon / @symoon94

Releases

No releases published

Packages

No packages published

Languages