A simple PyTorch implementation of digit recognition from NMIST database
All the scripts and data are in MNIST_PyTorch/.
To run the scripts, cd into MNIST_PyTorch/ first.
| Path | Info |
|---|---|
data/ |
All the data files |
digit_recognition_train.py |
Training script |
digit_recognition_test.py |
Testing script |
plot_loss.py |
Plot loss from training |
read_data.py |
Read data from data files |
*.pth |
Trained model(s) |
loss.json |
JSON data containing the loss values |
test*.log |
Logs from testing |