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
I'm trying load yolov5 as a torchscript model. I see that the output of the torchscript model ( I got using their export.py) has an output of type Tuple. I see that Triton supports Tuples for torchscript models but so far I only see for inputs. Does it work for output as well? Also, what should my config.pbtxt look like for the output?
P.S. I am experimenting with using the torchscript model only. I know it works fine with other backends like ONNX.
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
I'm trying load yolov5 as a torchscript model. I see that the output of the torchscript model ( I got using their export.py) has an output of type Tuple. I see that Triton supports Tuples for torchscript models but so far I only see for inputs. Does it work for output as well? Also, what should my
config.pbtxt
look like for the output?P.S. I am experimenting with using the torchscript model only. I know it works fine with other backends like ONNX.
Beta Was this translation helpful? Give feedback.
All reactions