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

Question

我有运行 TensorFlow 代码 1.x.It 可以工作 fine.Now 我已经生成了那个 code.They 的检查点也可以正常工作。现在我想运行我的 TensorFlow 2.x 文件中的这些检查点

OSError Traceback (most recent call last) in () 1 ----> 2 loaded = tf.saved_model.load("/content/model/") 3 4 5

1 frames /usr/local/lib/python3.6/dist-packages/tensorflow/python/saved_model/loader_impl.py in parse_saved_model(export_dir) 81 (export_dir, 82 constants.SAVED_MODEL_FILENAME_PBTXT, ---> 83 constants.SAVED_MODEL_FILENAME_PB)) 84 85

OSError: SavedModel file does not exist at: /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

Answer 1

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

Answer 2

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

例如tf.compat.v1.saved_model.load

您的情况可能值得一试：

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

Answer 3

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

新命令：

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

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

How to load Tensorflow 1.x saved model in TensorFlow 2.x?

nlp

python-3.x

tensorflow2.0