Can't load onnx model in TensorRT


I have an official model from , and I try to convert it into onnx model by using tutorial from

After finishing the conversion, I failed to load onnx model in TensorRT and received the error message:

Input filename: /path/to/onnx-model.onnx
ONNX IR version: 0.0.3
Opset version: 8
Producer name:
Producer version:
Model version: 0
Doc string:
While parsing node number 5 [Reshape]:
ERROR: builtin_op_importers.cpp:1558 In function importReshape:
[8] Assertion failed: get_shape_size(new_shape) == get_shape_size(weights.shape)

Here is my env:

  1. Linux version: Ubuntu 16.04
  2. GPU: RTX2070
  3. Nvidia driver version: 435.21
  4. CUDA version: 10.1
  5. CUDNN version: 7
  6. Mxnet version: 1.5.0
  7. TensorRT version:

Does anyone can help me to solve this error?