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