python-3.x - 遇到以下问题 : build_tensor_flow is not supported in Eager Mode

Question

我在玩 TensorFlow，我正在尝试将 Keras 模型导出为 TensorFlow 模型。我遇到了上述错误。我正在关注 Lynda ( https://www.linkedin.com/learning/building-deep-learning-applications-with-keras-2-0/exporting-google-cloud-compatible-models?u=42751868 ) 的“使用 Keras 2.0 构建深度学习应用程序”

在尝试构建 tensorflow 模型时，我遇到了这个错误，在第 66 行抛出，其中定义了添加元图和变量函数。

第 66 行，在 build_tensor_info 中
raise RuntimeError("在 Eager 模式下不支持 build_tensor_info。")
运行时错误:在 Eager 模式下不支持 build_tensor_info。

...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
        }
    )
...

大家有什么想法吗？

Answer 1

是因为您使用的是 tensorflow v2。您必须使用 tensorflow v2 兼容性并禁用 Eager 模式。

请注意您使用的 tensorflow 导入，例如，如果您使用 tensorflow_core，请确保您使用的是“tensorflow”中的所有依赖项。
您必须在代码之前添加:

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