README.md

MNIST Demo Project

Description

A demo project for MNIST CNN with PyTorch with training on multiple type of devices with PyTorch Lightning and an Django Rest Framework API deployable with Docker.

Setup with Docker

cd api && sudo docker-compose up --build

Training

cd models && python training.py train