Hello MNIST in 3 Frameworks
Simple MNIST classifier written in PyTorch, PyTorch Lightning, and Keras.
Install Dependencies
pip install -r requirements.txt
PyTorch / Lightning
# pytorch
python pytorch.py
# lightning
python pl_mnist.py
Keras
python keras.py