I am interested in computing the gradient of all parameters. Do you mean, I can use “loss.backward” in test to compute the gradient of parameters but I should not use “optimizer.step()” to update the model?