jshilong / FisherPruning

Group Fisher Pruning for Practical Network Compression(ICML2021)
152 stars 15 forks source link

FC layers of roi_head wasn't be pruned #13

Closed mls1999725 closed 2 years ago

mls1999725 commented 2 years ago

Hi, when I finetune the faster rcnn ( flops 50%, act 64% ), FC layers of roi_head wasn't be pruned, I wonder is it normal? image

jshilong commented 2 years ago

It is abnormal, two FCs have huge flops and they will be pruned with a high proportion. I guess there are some problems with your implementation

jshilong commented 2 years ago

Feel free to reopen the issue if there is any question