nn_pruning icon indicating copy to clipboard operation
nn_pruning copied to clipboard

Does this script run with the hybrid-filled method by default?Thanks!

Open iMountTai opened this issue 3 years ago • 1 comments

@madlag

iMountTai avatar Dec 13 '22 06:12 iMountTai

I had the same question. I understand I need to set "attention_pruning_method" : pruning_method and "dense_pruning_method" : pruning_method:1d_alt for hybrid pruning.

Is it that for hybrid-filled approach we train the model again with reinitializing the weights for heads which are not completely pruned and this is done separately after hybrid pruning? or Is there a argument I can set in the code for hybrid-filled pruning? @madlag @echarlaix

sangamesh-kodge avatar Jun 05 '23 07:06 sangamesh-kodge