Skip to content

Python scripts to generate unit level simple models with Tensoflow and Pytorch

License

Notifications You must be signed in to change notification settings

kumardesappan/ml-unit-test-models

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Models for ML inference SW unit test

  • 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

Steps to Install and Use

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

About

Python scripts to generate unit level simple models with Tensoflow and Pytorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages