onnx / onnx/models

[Announce] Next generation of ONNX Model Zoo

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

Nobody has claimed this yet.

announce enhancement
Dominant language
Jupyter Notebook
Stars
9.8k
Forks
1.6k
PR merge metrics
No merged PRs in 30d

Description

Feature Request

What is the problem that this feature solves?
  • All of existing ONNX Model Zoo models use opset_version<13, which is a quite old version (opset 12 released on May 8, 2020 in ONNX 1.7.0).
  • Existing models are hard to reproduce with outdated script ​
  • Same model usually has few versions​
  • No sufficient state-of-art models in the past 2 years
Describe the feature
  • Introduce new upload way with required reproduction files: https://github.com/onnx/models/pull/613
  • Move existing models (< opset_version 13) to models/legacy: https://github.com/onnx/models/issues/616
  • Add a few existing models from torch_hub or torchvision with opset_version 18 16: https://github.com/onnx/models/pull/614
  • Add more state-of-the-art models with opset_version 18 16: https://github.com/onnx/models/pull/615
  • Utilize benchmark tool (MLAgility) to verify uploaded models
  • Keep single version of model and going forward ONNX Model Zoo will regularly bump opset_version
  • New models directory hierarchy:
    • models/python: user-contribution-based models. There will be models along with reproduction script to ensure the models are reproducible.
    • models/mlagility: only contains models and test_data_set; reproduction script can be found at https://github.com/groq/mlagility/tree/main/models.
    • models/legacy: old models with opset_version < 13
  • ONNX Hub will support download models from the three source above
  • Web interface to view ONNX Model Zoo models
  • Eventually we will remove models under legacy
Relevant Model Zoo feature areas

Which area in the Model Zoo infrastructure does this impact?
Feature Area (e.g. CI, file organization, model storage, other):
Model storage/CI

Notes

Currently this feature is working in progress and we are developing in the "new-models" branch first. Ultimately we will move this branch into the main branch.

If anyone has any question/concern, feel free to raise in this thread. Thank you for the contribution.

cc @abock @ramkrishna2910

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

Start by reading the new-models branch and the linked work in PRs 613–615 and issue 616. Review the proposed models/python, models/mlagility, and models/legacy hierarchy, then inspect how ONNX Hub and MLAgility are expected to use them. Done would require the broader migration, reproduction workflow, benchmarking, and upload organization described here.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
ci-cd, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.