pytorch no_grad code example Example: with torch.no_grad() The wrapper "with torch.no_grad()" temporarily set all the requires_grad flag to false.