如何在 TensorFlow 2.x 中加载 TensorFlow 1.x 保存的模型？答案

【问题标题】：How to load Tensorflow 1.x saved model in TensorFlow 2.x?如何在 TensorFlow 2.x 中加载 TensorFlow 1.x 保存的模型？
【发布时间】：2020-05-06 05:39:58
【问题描述】：

我已经运行了 TensorFlow 1.x 的代码。它工作正常。现在我已经生成了该代码的检查点。它们也工作正常。现在我想在我的 TensorFlow 2.x 文件中运行这些检查点

() 中的 OSError Traceback（最近一次调用最后一次） 1 ----> 2 加载 = tf.saved_model.load("/content/model/") 3 4 5

1 帧 /usr/local/lib/python3.6/dist-packages/tensorflow/python/saved_model/loader_impl.py 在 parse_saved_model(export_dir) 81（出口目录， 82 个常量。SAVED_MODEL_FILENAME_PBTXT， ---> 83 个常量。SAVED_MODEL_FILENAME_PB)) 84 85

OSError：SavedModel 文件不存在于： /content/model//{saved_model.pbtxt|saved_model.pb}

这是错误代码已加载 = tf.saved_model.load("/content/model/")

分享您的看法我们如何解决此错误

This is my file hierarchy,how I uploaded files on colab."model" folder is containing ckpt files of machine learning model in tensorflow 1

【问题讨论】：

标签： python-3.x nlp tensorflow2.0

【解决方案1】：

我迟到了回答这个问题。但我认为我的回答可能会帮助其他正在寻找在 TensorFlow Version-2 上运行 TensorFlow Version-1 脚本的人。使用以下命令代替import tensorflow as tf

新命令：

import tensorflow.compat.v1 as tf
tf.compat.v1.disable_v2_behavior()

【讨论】：

【解决方案2】：

我不确定它在这种情况下是否有效，但有时添加 tf.compat.v1.[desired command] 可以使其在使用 tensorflow 2 时有效。

例如tf.compat.v1.saved_model.load

在你的情况下可能值得一试：

loaded=tf.compat.v1.saved_model.load("/content/model/")

【讨论】：

【解决方案3】：

我不认为这是可以做到的。 Tensorflow 2 在很大程度上与 Tensorflow 1 向后不兼容。

【讨论】：

“此外，SavedModels 或存储的 GraphDefs 将向后兼容。使用 1.x 保存的 SavedModels 将继续在 2.x 中加载和执行。”这是他们博客中的一行。 medium.com/tensorflow/… 我也在寻找这个问题的答案。在他们的博客中，他们说它将在加载保存的模型方面向后兼容。