Run into the following issue: build_tensor_flow is not supported in Eager Mode

Question

I am playing around with TensorFlow, and I am trying to export a Keras Model as a TensorFlow Model. And I ran into the above-mentioned error. I am following the "Build Deep Learning Applications with Keras 2.0" from Lynda (https://www.linkedin.com/learning/building-deep-learning-applications-with-keras-2-0/exporting-google-cloud-compatible-models?u=42751868)

While trying to build a tensor flow model, I came across this error, thrown at line 66 where the add meta graphs and variables function is defined.

line 66, in build_tensor_info raise RuntimeError("build_tensor_info is not supported in Eager mode.") RuntimeError: build_tensor_info is not supported in Eager mode.

...model_builder.add_meta_graph_and_variables(
        K.get_session(),
        tags=[tf.compat.v1.saved_model.tag_constants.SERVING],
        signature_def_map={
            tf.compat.v1.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY: signature_def
        }
    )
...

Any thoughts folks?

Cristian Zumelzu · Accepted Answer · 2020-04-30T08:32:34.387

12

Is because you are using tensorflow v2. You have to use the tensorflow v2 compatibility and disable eager mode.

Be careful with the tensorflow imports that you use, for example if you use tensorflow_core, be sure that you are using all the dependencies from "tensorflow". You have to add before your code:

import tensorflow as tf
if tf.executing_eagerly():
   tf.compat.v1.disable_eager_execution()

edited Apr 30 '20 at 08:32

answered Apr 29 '20 at 16:50

Cristian Zumelzu

842
10
15

Thanks, barring some tweaking for the indentation this solution worked for me! – Kris Stern Nov 26 '20 at 08:53

score 5 · Answer 2 · answered Oct 26 '20 at 02:26

Ran into that exact same problem when compiling that code from LinkedIn Learning for exporting the Keras model that was written using TensorFlow 1.x API. After replacing everything with the equivalent tf.compat.v1 functions, such as changing

model_builder = tf.saved_model.builder.SavedModelBuilder("exported_model")

to

model_builder = tf.compat.v1.saved_model.Builder("exported_model")

and disables eager execution as suggested above by Cristian Zumelzu, the code was able to run fine with the expected warnings about the deprecated functions.

Run into the following issue: build_tensor_flow is not supported in Eager Mode

2 Answers2