TensorRT-SSD icon indicating copy to clipboard operation
TensorRT-SSD copied to clipboard

PreluPlugin constructor from nvinfer1::Weights

Open GarrickLin opened this issue 7 years ago • 3 comments

I found that PreluPlugin still need a constructor for nvinfer1::Weights to init prelu parameters

GarrickLin avatar Feb 07 '18 08:02 GarrickLin

You can look at the tensorrt3.0 document.There is an introduction in the document.

chenzhi1992 avatar Feb 08 '18 00:02 chenzhi1992

Thanks for your contribution to TensorRT implementation @chenzhi1992 , I found that TensorRT3 already implemented createPReLU INvPlugin * createPReLUPlugin(float negSlope); INvPlugin * createPReLUPlugin(const void * data, size_t length); However it only supports channel_shared mode, so what we need to do is to extend it to cross_channel mode, is it right?

nyyznyyz1991 avatar Mar 29 '18 03:03 nyyznyyz1991

it it not a real PReLU as caffe, it is a leakyrelu with constant neg-slop.

GarrickLin avatar Mar 29 '18 04:03 GarrickLin