Simple Neural Network for digits recognition with Pytorch.
main.py: file main that Download the MNIST dataset (in a folder named input, automatically created in the project folder). After the download a Neural Network is trained and a simple test on an image is executed.
networks.py: file that contains some Neural network definition. These Neural Networks can be utilized in the main.py file.
.gitignore: gitognore for windows so, python language and vscodeIde.
pytorch installed with the command: pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
The requirements file automatically set the link for the right torch installation