Hello MNIST in 3 Frameworks

Simple MNIST classifier written in PyTorch, PyTorch Lightning, and Keras.

Install Dependencies

pip install -r requirements.txt

# pytorch
python pytorch.py

# lightning
python pl_mnist.py

python keras.py

Name		Name	Last commit message	Last commit date
Latest commit History 29 Commits
README.md		README.md
keras.py		keras.py
pl_cifar10.py		pl_cifar10.py
pl_mnist.py		pl_mnist.py
pytorch.py		pytorch.py
requirements.txt		requirements.txt