enable_tensor_checker

paddle.amp.debugging. enable_tensor_checker ( checker_config )

enable_tensor_checker(checker_config) 是开启模型级别的精度检查,与 disables_tensor_checker() 一起使用,通过这两个 API 的组合实现模型级别的精度检查,检查指定范围内所有算子的输出 Tensor。

参数

  • checker_config: TensorCheckerConfig 类型的参数,用于设置精度检查工具的调试选项。

返回

无返回值

注解

如果在 backward() 之前调用 disable_tensor_checker(),则不会检查梯度算子; 如果在 optimizer.step() 之前调用 disable_tensor_checker(),则不会检查优化器和其他与权重更新相关的算子。

代码示例

COPY-FROM: paddle.amp.debugging.enable_tensor_checker