New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Alobugdays - add dynamic axes onnx option #287

Merged

Data-Iab merged 1 commit into alobugdays from dynamic-onnx

Nov 25, 2022

Collaborator

Data-Iab commented Nov 18, 2022 •

edited

Loading

New feature : ONNX dynamic inputs exporter #260

# While overriding the exporter class
def __init__(dynamic: bool = False, **kwargs):
     if dynamic:
            # Keys are inputs names. Lists are indexes of axes that we want to set as dynamic.
            self.dynamic_axes = {"input1": [1, 2, 3], "input2":[1, 2, 3]}
    #....

This pull request includes

Bug fix (non-breaking change which fixes an issue)
New feature (non-breaking change which adds functionality)
Breaking change (fix or feature that would cause existing functionality to not work as expected)
This change requires a documentation update


          add dynamic axes onnx option

3f2e982

Data-Iab linked an issue

that may be closed by this pull request

ONNX dynamic inputs exporter #260

Open

Data-Iab added enhancement alonet alobugdays labels

anhtu293 self-requested a review

November 18, 2022 09:36

Data-Iab self-assigned this

Data-Iab merged commit 24b5ed3 into alobugdays

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

alobugdays alonet enhancement