shouxieai/tensorRT_Pro

WIN10 CUDA10.2 TRT7.0 出现错误

QIANXUNZDL123 opened this issue · 3 comments

QIANXUNZDL123 commented

我下载了最新的trt_pro的程序，使用了里面的exmple_simple_yolo，编译的时候没有出现问题，可以正常使用，但是使用自己模型就会出现问题：
image

我已经导出为动态的bs，pytorch=1.10，请问这是什么原因，我设置了绝对路径也是包这样的错误。。。

shouxieai commented

这是因为resize节点有问题，你需要降低pytorch版本，或者使用simplify过一下

QIANXUNZDL123 commented

这是因为resize节点有问题，你需要降低pytorch版本，或者使用simplify过一下

好的 ，我试试simplify，但是我在TensorRT主文件夹里面使用app_yolo编译我的模型是没有问题得，很奇怪

lwh1229 commented

这是因为resize节点有问题，你需要降低pytorch版本，或者使用simplify过一下

你好，我是使用的tensort7.0版本，我之前的onnx文件已经经过simplify了，报错提示节点有问题，然后我使用提供的脚本进行resize转换后变成了unsample，但onnx文件就无法解析了，请问改怎么处理