Project-MONAI / Project-MONAI/model-zoo

Support TRT compiler test in the CI test.

Open
#643 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
338
Forks
106
PR merge metrics
No merged PRs in 30d

Description

Is your feature request related to a problem? Please describe.
Since the MONAI core has introduced the TRT compiler feature after 1.4 and the model zoo has added a few bundles with the feature support. The previous CI test for the TensorRT conversion is not able to cover all cases. Therefore, we need to add a new CI test for this.

Describe the solution you'd like
A clear and concise description of what you want to happen.

Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.

Additional context
Add any other context or screenshots about the feature request here.

Contributor guide

Open the contributing guide

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 reviewing ci/bundle_custom_data.py around line 59 and the existing TensorRT conversion CI test. Identify the model-zoo bundles using the MONAI TRT compiler and determine how the new CI test should cover them. Done means the CI test exercises all relevant TRT compiler cases.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
ci-cd, machine-learning, testing
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.