Skip to content

Dynamic ONNX shape support#181

Merged
QueensGambit merged 1 commit intomasterfrom
onnx_dynamic_shape
Jul 25, 2022
Merged

Dynamic ONNX shape support#181
QueensGambit merged 1 commit intomasterfrom
onnx_dynamic_shape

Conversation

@QueensGambit
Copy link
Owner

This PR allows dynamic shapes of ONNX models. This avoids storing multiple onnx files for each batch size but with the same network parameters.

  • update TensorrtAPI
  • update OpenVinoAPI: (now needs to be openvino_2022.1.0 or higher!)

The required onnx files should use -1in the batch-size dimension and no longer have batch-size specifier in its file name, e.g. model-1.20787-0.581-0535-v3.0.onnx instead of model-1.20787-0.581-0535-v3.0-bsize-64.onnx.

* update TensorrtAPI
* update OpenVinoAPI: (now needs to be openvino_2022.1.0 or higher!)
@QueensGambit QueensGambit merged commit 593fcae into master Jul 25, 2022
@QueensGambit QueensGambit deleted the onnx_dynamic_shape branch July 29, 2022 17:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant