26、TensorFlow调试与TPU使用指南

TensorFlow调试与TPU使用指南

一、TensorFlow调试（tfdbg）

TensorFlow调试器（tfdbg）在高层的工作方式与其他流行的调试器（如pdb和gdb）类似。使用调试器的一般过程如下：
1. 设置断点 ：在代码中你想要暂停并检查变量的位置设置断点。
2. 以调试模式运行代码 ：启动代码的调试模式。
3. 检查代码 ：当代码在断点处暂停时，检查代码并继续下一步。

部分调试器还允许你在代码执行过程中交互式地观察变量，而不仅仅是在断点处。

要使用tfdbg，可按以下步骤操作：
1. 导入模块并包装会话 ：

from tensorflow.python import debug as tfd
with tfd.LocalCLIDebugWrapperSession(tf.Session()) as tfs:

2. 附加过滤器 ：为会话对象附加一个过滤器，这类似于在其他调试器中设置断点。例如，以下代码附加了一个 tfdbg.has_inf_or_nan 过滤器，当任何中间张量包含 nan 或 inf 值时，代码将暂停：