gridai / hello_mnists

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About


Languages

Language:Python 100.0%