tensorflow - 如何在新的 Colab 笔记本中加载 SavedModel？

Question

我正在使用谷歌 Colab。在第一个笔记本中，我使用以下方法保存了我的模型：model.save('my_model.h5'). 在同一个笔记本中，我可以看到模型已存储。输入：ls -d $PWD/* 输出：/content/my_model.h5

在同一个笔记本中，我可以使用以下方法恢复保存的模型： model = tf.keras.models.load_model('my_model.h5')

但是如果我想在新笔记本中加载相同的模型，我会收到一条错误消息。输入：

from keras.models import load_model
new_model = tf.keras.models.load_model('/content/my_model.h5')

错误信息：

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

谢谢你的帮助，

鲍里斯

标签： tensorflowgoogle-colaboratory

Answer 1

每个笔记本都在不同的机器上。您需要使用 drive.mount("drive") 将其保存在谷歌驱动器上。然后，您可以在不同的笔记本上共享模型。