tensor flow eager mode

does habana support tensor flow eager mode? do I need to turn it off?

Hi Jonathan,

thanks for your question and interest in Habana. We recommend use of TensorFlow graph mode. By default, models will run in graph mode if you are using a Tensorflow 2.x training script using Keras or wrapped in tf.function. For models still using tf.Session, it is recommended to disable eager execution using the function call tf.compat.v1.disable_eager_execution(). Please refer to the Habana Migration guide for more information including a detailed table on how to manage the model execution mode in single card or in multi-card configuration using Horovod.

1 Like