onnx / onnx/models

Can we include some quantized models w.r.t. Quantization-aware Training?

Open
#362 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

model-request
Dominant language
Jupyter Notebook
Stars
9.8k
Forks
1.6k
PR merge metrics
No merged PRs in 30d

Description

Model Request

Which model would you like to see in the model zoo?

A quantized MobileNet (doesn't matter which version) could be fine. TensorFlow has published end to end quantized MobileNetV1.

Describe why this model is relevant

I am interested in the quantization representation of ONNX, and how it peserves quantization parameters which are generated by frameworks during quantization-aware training. I have read ONNX documents and operator spec, but a quantized model could have great impact to help people understand ONNX quantization. It won't be limited to MobileNet I think.

Model information

Source repository (e.g. Github link): TFLite quantized MobileNetV1 doc.
Model Zoo category (e.g. Vision): Vision or any.
Model Zoo sub-category (e.g. Image Classification): Image Classification or any.

Please feel free to ping me if any addition information is needed.

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

No repository file or test is named. Start by reviewing the model-zoo contribution structure and the linked TensorFlow quantized MobileNetV1 source, then determine how an ONNX model should preserve quantization-aware-training parameters. Done means a validated quantized MobileNet model is added to an appropriate vision classification category.

Written by the indexing model from the issue text.

Assessment

Tech stack
tensorflow
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.