Pytorch Mnist test

This is a deep learning test, using Pytorch, to train a model to recognize Mnist images.

Run

To run this project you need to install Python 3.7, Pytorch, and nvidia cuda.

Then you can launch the training:

python train.py

Or test the pretrained model:

python test.py