[ 参数完全一致 ] torch.set_grad_enabled

torch.set_grad_enabled

torch.set_grad_enabled(mode)

paddle.set_grad_enabled

paddle.set_grad_enabled(mode)

两者功能一致,参数完全一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
mode mode 启用(True)或禁用(False)动态图梯度计算。