You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Many ONNX models have dynamic input shapes such as 3xHxW that we cannot load.
Due to our static computational graph we cannot support these dynamic shapes, so I propose that ONNX should ask for the input shape if this is indeed required (dynamic input). Else, nothing.
Examples:
// Not dynamic input shape => Okay
Net *net = import_net_from_onnx_file(model_path);
// Dynamic input shape + No given input shape => Error message asking for an input shape
Net *net = import_net_from_onnx_file(model_path);
// Dynamic input shape + Given input shape => Okay
Net *net = import_net_from_onnx_file(model_path, {in_channels, in_height, in_width});
The text was updated successfully, but these errors were encountered:
salvacarrion
changed the title
Load dynamic inputs ONNX
Load dynamic inputs shapes from ONNX
Dec 1, 2021
Many ONNX models have dynamic input shapes such as
3xHxW
that we cannot load.Due to our static computational graph we cannot support these dynamic shapes, so I propose that ONNX should ask for the input shape if this is indeed required (dynamic input). Else, nothing.
Examples:
The text was updated successfully, but these errors were encountered: