pytorch 转onnx 到tensorrt 坑
pytorch 版本1.6tensorrt 版本7.21、 模型中含有 ConvTranspose2d 上传采样时, tensorrt 7.0 版本 导出的模型,输出结果会和pytorch不一样,经测试tensorrt7.2版本可以 结果近似一致,也不是很准,由于tensort核心没有开源,无法探究,尽量不要是用Upsample,因为这个层在tensorrt7.2版本中还不支持,写起插件来很麻烦,建议不要使用2、还是 pytorch ConvTranspose2d上采样,导出onnx模型在op.
原创
2020-10-24 14:51:39 ·
1948 阅读 ·
2 评论