The only reason I am trying v 0.2 is that the 0.4 version seems to produce more weird errors!: I though it would be a better idea to try a lower version in order to not face with these errors.
If I start using a 0.4 version, this is the error I get :
  File "prune.py", line 79, in <module>
    mask = weight_copy.abs().gt(thre).float().cuda()
RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #2 'other'