这实际是由于pytorch与TensorRT版本不匹配,我的TensorRT是7.0,pytorch应该是1.4,但我用了1.7. 因此需要用1.7重新读取权重文件,然后用老的方式保存,再用onnx导出
確定! 回上一頁