jwillette
(Jeff Willette)
October 11, 2020, 11:20am
1
I was looking at the Pytorch pruning tutorial (https://pytorch.org/tutorials/intermediate/pruning_tutorial.html# ) and I am left confused about where in the process this pruning happens.
Does the setup in the tutorial happen before training, and then it is trained with the pruning method somewhere in the loop
Do the strategies in the tutorial happen after training when we have the final network weights?
albanD
(Alban D)
October 11, 2020, 5:07pm
2
Hi,
This tutorial shows how to use pruning for the weights that are used during training.
It is not a pre or post processing technique.
In particular, the prunning happen when you access the weights where you registered the prunning, they will automatically get prunned.
jwillette
(Jeff Willette)
October 12, 2020, 1:57am
3
Thanks, so just to verify, I would follow the tutorial, and then make a normal training loop without any other modification and the model will automatically be pruned while training?