Why weights are not masked-out

[thomascong121]

trafficstars

Hi, Thanks for the great work. I have a question with regard to your gradual prune implementation. I notice that in your implementation, you commented out the code for masking out the model weights (line 88 in utils/prune.py). Is there a particular reason for doing this?