cekong

导航

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的替代)

 

posted on 2019-07-15 08:58  cekong  阅读(601)  评论(0编辑  收藏  举报