禁用`@tf.function` 装饰器进行调试？

Question

在 TensorFlow 2 中，@tf.function装饰器允许 Python 函数变成 TensorFlow 图（或多或少），并且可以带来一些性能改进。但是，当以这种方式修饰时，Python 不再在每次运行时跟踪这些函数。这使得使用 Python 调试器调试函数变得更加困难。有没有办法@tf.function暂时禁用所有装饰器以方便调试？

Answer 1

使用tf.config.run_functions_eagerly(True).