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

Question

我已經運行了 TensorFlow 1.x 的代碼。它工作正常。現在我已經生成了該代碼的檢查點。它們也工作正常。 現在我想在我的 TensorFlow 2.x 文件中運行這些檢查點

OSError Traceback（最近一次調用最后一次） in () 1 ----> 2 loaded = tf.saved_model.load("/content/model/") 3 4 5

1 幀 /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)_FILE 84 85

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

這是錯誤代碼加載 = tf.saved_model.load("/content/model/")

分享您的觀點 我們如何解決此錯誤

這是我的文件層次結構，我如何在 colab 上上傳文件。“模型”文件夾包含 tensorflow 1 中機器學習模型的 ckpt 文件

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()