Commit Graph
1 Commits
Author SHA1 Message Date
044daafd9d Add output_names argument for ONNX export with dynamic axes (#3456)
* 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

* use first output only + cleanup

Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-04 22:01:58 +02:00