Some question about the code

[RoyalVane]

PolarizationPruning/imagenet/main.py

Line 878 in f19f3fd

m.weight.grad.data.add_(sparsity * torch.sign(m.weight.data))

[RoyalVane]

Dear authors,

Thank you for your awesome work! I have a question about the code (line878 of PolarizationPruning/imagenet/main.py)

In my opinion, the updateBN should be "m.weight.data.add_(sparsity * torch.sign(m.weight.grad.data))"
I would appreciate it if you can solve my confusion.

Thank you!