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

Question

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

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

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

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}

谢谢你的帮助，

鲍里斯

Answer 1

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