Do not ignore output shapes when building ONNX InferenceModel

EmbarkStudios / cervo

Utility wrappers for tract

Apache License 2.0

40 stars 1 forks source link

Do not ignore output shapes when building ONNX InferenceModel #51

Closed klashenriksson closed 10 months ago

klashenriksson commented 10 months ago

Description of Changes

We ran in to an issue where a onnx-based InferenceModel did not contain any output-shape information for onnx models with dynamic output shapes. This change fixes this issue by specifying to not ignore output shapes before building the InferenceModel.