search

chenzhi1992 / TensorRT-SSD

Use TensorRT API to implement Caffe-SSD， SSD（channel pruning）， Mobilenet-SSD
251 stars 84 forks source link

Why we need the Softmax Plugin layer? #3

Closed Orangels closed 6 years ago

Orangels commented 6 years ago

Hi, thanks a lot about the code .I found tensorRT had implemented the softmax layer.Do we still need to implement softmax plugin ourselves?