chenzhi1992 / TensorRT-SSD

Use TensorRT API to implement Caffe-SSD, SSD(channel pruning), Mobilenet-SSD
251 stars 84 forks source link

SSD channel pruning #9

Open GraphicsHunter opened 6 years ago

GraphicsHunter commented 6 years ago

Hi. How do you implement channel pruning for SSD? Is the channel pruning done during the training or during the inference?

chenzhi1992 commented 6 years ago

TRAINING @tianfanzhu