Open shadowpa0327 opened 1 year ago
Hi @shadowpa0327, the importance module may not be always robust for all models. we will investigate the issue further to improve its reliability. Thank you for the issue.
If it is an emergency to prune this model, you can quickly implement your own importance estimator following the tutorial provided in this link: https://github.com/VainF/Torch-Pruning/blob/master/tutorials/1%20-%20Customize%20Your%20Own%20Pruners.ipynb.
Thanks for the timely response! I will give it a try and try to figure the factor behind this error. If you needed I can also provide a script to reproduce this issue.
Description
Hi! Currently, I try to apply
MagnitudePruner
onto my model withtp.importance.MagnitudeImportance
as importance metrics. There result in some error while calculating score as shown below:I have traced part of the code, it seems that
group_imp
is empty, can you provide me some tips to debug? Thank you!Reference
Model I try to prune:
https://github.com/datvuthanh/HybridNets