opset11

将yolov8-face里的模型导出到指定opset11

https://github.com/derronqi/yolov8-face解决办法:1.导出前1报错内容(Tensorinput,Tensorweight,Tensorbias,tupleofintsstride,tupleofintspadding,tupleofintsdilation,intgroups)didn'tmatchbecausesomeoftheargumentshavein
we34dfg·2023-12-02 08:26

pytorch模型转tensorrt，pad报错，opset11

pytorch模型转tensorrt，pad报错目录背景软件环境问题定位解决思路解决方案1.在pytorch中不使用pad2.转onnx时使用opset93.修改onnx-tensorrt源码4.自定义torchpad层实现背景工程化模型部署时，对pytorch模型使用tensorrt进行推理加速，需要将pytorch转为onnx进而转化为tensorrt进行部署。但是当pytorch的基本模型中
bmfire·2023-01-02 12:21
上一页 1 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他