TensorFlow Graphics 调试模式

Tensorflow Graphics 严重依赖于 L2 归一化张量,以及要求其输入处于特定范围内的三角函数。在优化过程中,更新可能会使这些变量取值导致这些函数返回 InfNaN 值。为了简化此类问题的调试,TensorFlow Graphics 提供了一个调试标志,该标志会向图中注入断言以检查正确的范围和返回值的有效性。由于这可能会减慢计算速度,因此调试标志默认设置为 False

用户可以设置 -tfg_debug 标志以在调试模式下运行他们的代码。还可以通过首先导入这两个模块以编程方式设置标志

from absl import flags
from tensorflow_graphics.util import tfg_flags

然后将以下行添加到代码中。

flags.FLAGS[tfg_flags.TFG_ADD_ASSERTS_TO_GRAPH].value = True