Can u please show how to use with torch.no_grad():
And replace volatile
I am testing a GitHub project and they have used volatile =False as a function argument
So do I need to replace it with requires_grad=True