i understood that
-
eval() changes the bn and dropout layer’s behaviour
-
torch.no_grad() deals with the autograd engine and stops it from calculating the gradients, which is the recommended way of doing validation
BUT, I didnt understand the use of with torch.set_grad_enabled()
Can you pls explain what is its use and where exactly can it be used.
Thanks ! 