Skip to content
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

Add output_names argument for ONNX export with dynamic axes #3456

Merged
merged 3 commits into from
Jun 4, 2021
Merged

Add output_names argument for ONNX export with dynamic axes #3456

merged 3 commits into from
Jun 4, 2021

Commits on Jun 4, 2021

  1. Add output names & dynamic axes for onnx export 

    Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3
    Samridha Shrestha committed Jun 4, 2021
    Configuration menu
    Copy the full SHA
    e7cf148 View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'upstream/develop' into develop

    Samridha Shrestha committed Jun 4, 2021
    Configuration menu
    Copy the full SHA
    f256456 View commit details
    Browse the repository at this point in the history

  3. use first output only + cleanup

    @glenn-jocher
    glenn-jocher committed Jun 4, 2021
    Configuration menu
    Copy the full SHA
    7b79da1 View commit details
    Browse the repository at this point in the history