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.
cd api && sudo docker-compose up --buildcd models && python training.py trainmbml84/mnist-pytorch
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|