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

时间:2020-04-04 09:10:41

标签: tensorflow google-colaboratory

我正在使用Google 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}

感谢您的帮助,

鲍里斯

1 个答案:

答案 0 :(得分:0)

每个笔记本电脑都在不同的计算机上。您需要使用drive.mount(“ drive”)将其保存在Google驱动器上。然后,您可以在不同的笔记本电脑上共享模型。