set_grad_enabled

paddle. set_grad_enabled ( mode )

创建启用或禁用动态图梯度计算的上下文。

参数

  • mode (bool) - 启用(True)或禁用(False)动态图梯度计算。

返回

None

代码示例

COPY-FROM: paddle.set_grad_enabled