Skip to content

v0.1.0

Latest
Compare
Choose a tag to compare
@arshan-ritual arshan-ritual released this 29 Mar 16:44
03ebc6f

Initial release of simple-ml-models: lightweight torch & onnx models to showcase infernet's workflows.

Features

  • Iris Classification: A simple project that trains a flower classification model on the Iris Dataset, and exports it in both pytorch and onnx format. Model files are hosted on huggingface.