pytorch1.5 无法使用plugin_onddnx_export.py 导出模型

dlunion / tensorRTIntegrate

TensorRT ONNX Plugin、Inference、Compile

464 stars 114 forks source link

Open justinyym opened 4 years ago

justinyym commented 4 years ago

使用最新版的1.5无法导出模型，但是1.2可以导出，请问两个版本之间的差异大概是什么？是不是函数用的编写方式做了更改

hopef commented 4 years ago

是的，我测试使用的是pytorch1.4，还是ok的。onnx这块儿，pytorch改的很频繁

KangolHsu commented 3 years ago

same problem. To be solved