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

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •