pytorch with torch.no_grad() code example

Example: with torch.no_grad()

The wrapper "with torch.no_grad()" 
temporarily set all the requires_grad flag to false.