torch.no_grad

PyTorch里的requires_grad、volatile及no_grad

requires_grad=True 要求计算梯度

requires_grad=False 不要求计算梯度

with torch.no_grad()或者@torch.no_grad()中的数据不需要计算梯度,也不会进行反向传播

torch.no_grad()是新版本pytorch中volatile的替代)

你可能感兴趣的:(torch.no_grad)