/mnist-pytorch

A demo project for MNIST CNN with PyTorch

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

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