Hi, I’m wondering why you added ‘torch.no_grad()’ at the top. Is it to prevent the parameters from updating? So if I wish to train the parameters aftering adding the gaussian noise, should I not include the no grad argument?
Thanks!