Skip to content

mbml84/mnist-pytorch

Repository files navigation

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