Ask Your Question
0

I am able to convert pre-trained models(pfe.onnx and rpn.onnx) into tensorrt. But I am not able to convert our models into tensorrt.

asked 2019-07-16 00:26:30 -0500

Siva gravatar image

ONNX IR version: 0.0.4 Opset version: 9 Producer name: pytorch Producer version: 1.1 Domain: Model version: 0 Doc string: While parsing node number 16 [Squeeze -> “175”]: ERROR: /home/erisuser/p4sw/sw/gpgpu/MachineLearning/DIT/release/5.0/parsers/onnxOpenSource/builtin_op_importers.cpp:1570 In function importSqueeze: [8] Assertion failed: axis != BATCH_DIM ERROR: failed to parse onnx file Please address the issue. Thanks in advance

edit retag flag offensive close merge delete

1 Answer

Sort by » oldest newest most voted
0

answered 2019-07-16 01:01:50 -0500

kosuke_murakami gravatar image

Thank you for your question. Unfortunately, if you want to use your own model, it is the out of the scope of Autoware.

You can ask a question regarding TensorRT here. https://www.nvidia.com/object/support...

Here for pytorch questions. https://discuss.pytorch.org/

edit flag offensive delete link more

Comments

Thanks for the reply. May I know which version of Pytorch you used to convert to onnx models and tensorrt version?

Siva gravatar imageSiva ( 2019-07-16 01:08:24 -0500 )edit

Please check the readme.md for TensorRT version. https://gitlab.com/kosuke_m/core_perc...

Any version of pytorch which can convert to onnx model should be fine.

kosuke_murakami gravatar imagekosuke_murakami ( 2019-07-16 03:36:10 -0500 )edit

Got it,

I am using Pytorch 1.1,tensorrt 5.0(as you mentioned in readme), ONNX IR version:0.0.4, Opset version:9 and converted to onnx. Now i can able to convert rpn.onnx to rpn.trt but i am not able to convert pfe.onnx to pfe.trt.

The error is like below:

While parsing node number 16 [Squeeze -> “175”]: ERROR: /home/erisuser/p4sw/sw/gpgpu/MachineLearning/DIT/release/5.0/parsers/onnxOpenSource/builtin_op_importers.cpp:1570 In function importSqueeze: [8] Assertion failed: axis != BATCH_DIM ERROR: failed to parse onnx file

Siva gravatar imageSiva ( 2019-07-16 07:09:01 -0500 )edit

Your Answer

Please start posting anonymously - your entry will be published after you log in or create a new account.

Add Answer

Question Tools

Stats

Asked: 2019-07-16 00:26:30 -0500

Seen: 117 times

Last updated: Jul 16