huggingface / nn_pruning

Prune a model while finetuning or training.
Apache License 2.0
394 stars 58 forks source link

unable to replicate results of example notebooks #34

Open rohanshingade opened 2 years ago

rohanshingade commented 2 years ago

Can you please provide a dockerfile or requirement.txt to reproduce the results. I installed this library but had to make a few changes to run the example notebooks. On running the notebook, the number of parameters were the same as the original model. No pruning was done.