在Python2中,@tf.function装饰器允许TensorFlow函数变成TensorFlow图(或多或少),并且可以带来一些性能改进。但是,当以这种方式装饰时,Python no longer traces the functions each time they run。这使得使用Python调试器调试函数变得更加困难。有没有一种方法可以暂时禁用所有@tf.function装饰器以方便调试?
在进阶示例中使用了一个tensorflow.keras子类。@tf.function装饰器在train_step和test_step上的存在意味着模型在graph模式下执行(不确定这是否是正确的术语,我的意思是oposite to eager模式)。如果去掉这些装饰器,我就可以直接进入模型call函数,并查看每个层的输入/输出张量,这是很整洁的。
我的问题是,是否有一种以编程方式启用/禁用@tf.function装饰器</e