set_grad_enabled¶

paddle. set_grad_enabled ( mode ) [源代码] ¶

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

参数¶

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

返回¶

None

代码示例¶

          >>> import paddle
>>> x = paddle.to_tensor([1.], stop_gradient=False)
>>> is_train = False
>>> with paddle.set_grad_enabled(is_train):
...     y = x * 2
>>> print(y.stop_gradient)
True

>>> paddle.set_grad_enabled(True)
>>> y = x * 2
>>> print(y.stop_gradient)
False

>>> paddle.set_grad_enabled(False)
>>> y = x * 2
>>> print(y.stop_gradient)
True

         