- Python scripts to generate unit level simple models with Tensoflow and Pytorch
- Export the models in TFLite or ONNX format
- Test the inference of the exported models using TFLite interpreter or ONNX runtime
Theese examples are exapected to work on Ubuntu >= 18.04 and MacOS >= 11.4
git clone https://github.com/kumardesappan/ml-unit-test-models.git
cd ml-unit-test-models
pip3 install -r ./requirements.txt