如何在新的 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。