Can't load onnx model in TensorRT 5.1.5.0

Hi~

I have an official model fromhttps://github.com/onnx/models/tree/master/vision/body_analysis/arcface , and I try to convert it into onnx model by using tutorial from https://mxnet.incubator.apache.org/versions/master/tutorials/onnx/export_mxnet_to_onnx.html

After finishing the conversion, I failed to load onnx model in TensorRT 5.1.5.0 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:
Domain:
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: 5.1.5.0

Does anyone can help me to solve this error?

Thanks.