Do I just need to implement the softmax layer ?

[ljq19950723]

Firstly,thank you for your code.And I compared the pluginImplement.cpp of your code with the samplePlugin.cpp of TensorRT's samplePlugin.Do I only need to implement the softmax layer ?And the mobilenet can work on TensorRT.

[chenzhi1992]

You need to implement the depthwise layer， too.

[ljq19950723]

And Do I need to implement the class SoftmaxPlugin's functions that inherited from the parent class only?Or any other functions?

[chenzhi1992]

Yes， you need to implement the SoftmaxPlugin

[xchani]

I found the channels of output blobs from plugin reshape layer is actually the number of classes, hence off-the-shelf across-channel softmax layer can satisfy demand. @chenzhi1992

[chenzhi1992]

Allright, you can verify that the result is correct.

[twmht]

@chenzhi1992

Do you have plan to release the code of softmax and depthwise layer?

[linux-devil]

@xchani Did off-the-shelf across-channel softmax worked for you ?

[xchani]

@linux-devil NO. Though off-the-shelf softmax and my own implemented softmax layer both operate across-channel, off-the-shelf softmax still produce wrong results. I cannot figure it out why it happends.

[Optimus1072]

@xchani Did it work with your softmax implementation. Can you please share it and explain why the dimension of last layer is 12764?.
Thanks :-)