Hello MNIST in 4 frameworks

PyTorch / Lightning

pip install -r requirements/pytorch.txt

# pytorch
python pytorch.py

# lightning
python pytorch_ligtning_mnist.py

pip install -r requirements/tensorflow.txt

# keras
python keras.py

# tensorflow
python tf2.py

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
requirements		requirements
README.md		README.md
keras.py		keras.py
pytorch.py		pytorch.py
pytorch_lightning_mnist.py		pytorch_lightning_mnist.py
tf2.py		tf2.py