无法使用更新代码在 tensorflow V2 中加载 Tensorflow v1 模型？

Question

我保存了 .ckpt 和 .h5 格式的模型，我的文件夹包含 4 个文件。

如果我使用命令加载版本 1

model.load("model.ckpt") or model.load("model.h5")

在 Tensorflow 版本 1 中加载成功，但在 tensorflow 版本 2 中加载时显示错误

  tf.keras.models.load_model('model.h5') or  tf.keras.models.load_model('model.ckpt')

有时会报错

SavedModel file does not exist at: /model/model.ckpt/{saved_model.pbtxt|saved_model.pb}

或有时报错

OSError: Unable to open file (unable to open file: name = 'model.ckpt', errno = 2, error message = 
'No such file or directory', flags = 0, o_flags = 0)

我仔细检查了我在 google colab 中上传的路径，并给出了确切的位置，但无法正常工作。

这是我的代码link

https://colab.research.google.com/drive/1hf7AwEMJHf4zLZOBSzxejA2ISuTQ3N9K

Answer 1

如果文件位于驱动器的文件夹内，请在路径中添加文件夹的名称。否则按原样使用代码。

import os
from google.colab import drive
drive.mount('/content/gdrive')
model_file_name = os.path.join("/content/gdrive/My Drive", 'model.h5')# CHANGE MODEL NAME
model = tf.keras.models.load_model(model_file_name)

执行时会显示 link 单击它并选择文件所在的驱动器帐户，然后将密钥复制粘贴到您的程序中。

无法使用更新代码在 tensorflow V2 中加载 Tensorflow v1 模型？

can't load Tensorflow v1 model in tensorflow V2 with update code?

nlp

python-3.x

deep-learning

tensorflow2.0