14 lines (9 loc) · 298 Bytes

MNIST Model

How to Run

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

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

Export Model from Pytorch to ONNX

python export_to_onnx.py