Onnx-tensorrt: Why is there delay in providing support for Prelu?

Created on 15 Jul 2019  路  6Comments  路  Source: onnx/onnx-tensorrt

@yinghai @benbarsdell @kevinch-nv

Hello, we now have a project based on InsightFace and want to accelerate with TensorRT. However, InsightFace is based on the Mxnet model, so it needs to be converted to the Onnx model. Among them, the model requires a Prelu activation function, but TensorRT-Onnx does not support Prelu operation. What I want to know is, when will the TensorRT-Onnx project support Prelu under the Onnx model, is there a development plan? Thank you!

Most helpful comment

@flyunix onnx-tensorrt supports customized ops through plugins. Please take a look at https://github.com/onnx/onnx-tensorrt/blob/master/FancyActivation.cu for an example.

All 6 comments

@Yangqing @Maratyszcza @yinghai @nswamy @flyunix
Hello, we now have a project based on InsightFace and want to accelerate with TensorRT. However, InsightFace is based on the Mxnet model, so it needs to be converted to the Onnx model. Among them, the model requires a Prelu activation function, but TensorRT-Onnx does not support Prelu operation. What I want to know is, when will the TensorRT-Onnx project support Prelu under the Onnx model, is there a development plan? Thank you!

@flyunix onnx-tensorrt supports customized ops through plugins. Please take a look at https://github.com/onnx/onnx-tensorrt/blob/master/FancyActivation.cu for an example.

@yinghai no useful.

@yinghai no useful.

when will the TensorRT-Onnx project support Prelu under the Onnx model, is there a development plan? Thank you!

@flyunix PRELU is fully supported now in TensorRT 6.0. Please download the latest binaries and try importing your model. Closing this issue now, thanks!

@flyunix onnx-tensorrt supports customized ops through plugins. Please take a look at https://github.com/onnx/onnx-tensorrt/blob/master/FancyActivation.cu for an example.

The link is not reachable

Was this page helpful?
0 / 5 - 0 ratings

Related issues

andrew-yang0722 picture andrew-yang0722  路  3Comments

skmhrk1209 picture skmhrk1209  路  6Comments

aif2017 picture aif2017  路  3Comments

qigtang picture qigtang  路  8Comments

RizhaoCai picture RizhaoCai  路  7Comments