RMNet icon indicating copy to clipboard operation
RMNet copied to clipboard

没有采用mask卷积,如何用slimming方法进行剪枝

Open Serissa opened this issue 3 years ago • 0 comments

您好!我想对yolov5主干网络采用RM操作然后再采用slimming进行剪枝。我在网络中没有添加mask卷积操作(rmnet_pruning.py中采用),请问稀疏化训练是要添加在训练yolov5模型的时候还是在RM操作后再进行稀疏化训练,最后进行slimming进行剪枝?

Serissa avatar Feb 21 '22 03:02 Serissa