As I am pretty new to this modifying gradients.can you provide any example or reference for that?
I am planning to make the param.grad=0 for those parameter before optimizer.step() not sure weather it’s a correct approach or not