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