An implementation of LeNet-5 in pytorch, readily packaged to be trained on the MNIST data set.
To install, please run
pip install .
To get a trained ONNX model, please run e.g.
train-lenet5-mnist --output-path lenet5-mnist.onnx --num-epochs 100 --batch-size 64 --learning-rate 0.001 --random-seed 0
For a demonstration of how to use the trained model, please refer to the provided example.