Simple handwritten digits classification base on MNIST with Pytorch (with detailed annotation).
Python3
PyTorch == 2.0.0
torchvision == 0.15.1
$ git clone https://github.com/CENHM/Pytorch-MNIST.git
$ cd Pytorch-MNIST
$ python3 mnist.py
The program will download MNIST train & test datasets automatically.