is_grad_enabled

paddle. is_grad_enabled ( ) [源代码]

判断当前动态图下是否启用了计算梯度模式。

返回

bool,判断当前动态图是否启用了计算梯度的模式,若是则为 True,反之为 False。

代码示例

import paddle

# Dygraph gradient calculation mode is enabled by default.
paddle.is_grad_enabled() # True

with paddle.set_grad_enabled(False):
    paddle.is_grad_enabled() # False

paddle.enable_static()
paddle.is_grad_enabled() # False