tensorflow 调试

tensorflow 调试

tf.config.experimental_run_functions_eagerly(True) 是 TensorFlow 中的一个配置函数，它的作用是：

让 @tf.function 装饰的函数以 Eager 模式（即时执行）运行，而不是被编译成图（Graph）执行。

在 TensorFlow 2 中，推荐使用 @tf.function 来将 Python 函数转换为计算图，以获得更高性能（比如更快的训练、更好的跨设备兼容性等）。但是这样做会牺牲一部分调试的便利性，因为图模式下不能逐行调试、不能直接用 print()。

这时，如果你设置：

tf.config.experimental_run_functions_eagerly(True)

TensorFlow 会强制所有使用 @tf.function 装饰的函数 以 Eager 模式运行，就像普通的 Python 函数一样，便于调试。