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.
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
@Siva, did you manage to find a solution to your problem ?