mbml84 / mnist-pytorch

A demo project for MNIST CNN with PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

A demo project for MNIST CNN with PyTorch

License:GNU General Public License v3.0


Languages

Language:Python 96.5%Language:Dockerfile 3.2%Language:Shell 0.3%