alicera opened a new issue #18509: URL: https://github.com/apache/incubator-mxnet/issues/18509
cause; I convert mxnet model to onnx ,it is ok. But when I use `onnx2trt *.onnx -o *.trt` the model is fail env : pip install mxnet-cu102 pip install onnx==1.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple model ; https://www.dropbox.com/s/akxeqp99jvsd6z7/model-MobileFaceNet-arcface-ms1m-refine-v1.zip?dl=0 ref : https://github.com/deepinsight/insightface mxnet to onnx code; > ` import mxnet as mx import numpy as np from mxnet.contrib import onnx as onnx_mxnet sym = './model/model-symbol.json' params = './model/model-0000.params' input_shape = (1,3,112,112) onnx_file = './mxnet_exported_mobileface.onnx' converted_model_path = onnx_mxnet.export_model(sym, params, [input_shape], np.float32, onnx_file) print("finish ") > ` onnx2trt *.onnx -o *.trt Error; ` Parsing model [2020-06-08 04:45:16 ERROR] (Unnamed Layer* 10) [Parametric ReLU]: slope tensor must be unidirectional broadcastable to input tensor [2020-06-08 04:45:16 ERROR] (Unnamed Layer* 10) [Parametric ReLU]: slope tensor must be unidirectional broadcastable to input tensor [2020-06-08 04:45:16 ERROR] (Unnamed Layer* 10) [Parametric ReLU]: slope tensor must be unidirectional broadcastable to input tensor While parsing node number 5 [Conv -> "conv_2_dw_conv2d"]: ERROR: /opt/onnx-tensorrt/builtin_op_importers.cpp:463 In function importConv: [8] Assertion failed: nbSpatialDims == kernel_weights.shape.nbDims - 2 ` ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
