-
由 Sam_S 提交于
* 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>