FisherPruning icon indicating copy to clipboard operation
FisherPruning copied to clipboard

Resume pruning

Open Janspiry opened this issue 3 years ago • 0 comments

Hi, thanks for this awesome work. I'm wondering if there is an option to continue pruning, I tried setting pruning=True and deploy_from=checkpoint to achieve it, but it didn't work. And then I modify the code in https://github.com/jshilong/FisherPruning/blob/b28ea9185f1269ef4597303ccb6be3300c7e1b93/detection/tools/fisher_pruning.py#L113, making model loading mask and checkpoint when pruning. At the same time I comment the code in https://github.com/jshilong/FisherPruning/blob/b28ea9185f1269ef4597303ccb6be3300c7e1b93/detection/tools/fisher_pruning.py#L133 to prevent model sets mask to 1. Although it can obtain information about the pruned channels, it still prunes them according to the initial model.

Is there anything else I need to change?

Janspiry avatar Jun 02 '22 15:06 Janspiry