jshilong / FisherPruning

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

The batch size and number of gpu during pruning. #7

Closed mls1999725 closed 2 years ago

mls1999725 commented 2 years ago

Hi, I basically completed the pruning codes of faster-rcnn. I wonder the batch size and the number of gpu during pruning faster-rcnn, is it 16 on 8 2080ti which is same as all detection models? Thank you very much!