MNIST Model

How to Run

python3 -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt
python mnist.py

The code is from https://github.com/pytorch/examples/tree/main/mnist.

Export Model from Pytorch to ONNX

python export_to_onnx.py