我正在使用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}
感谢您的帮助,
鲍里斯
答案 0 :(得分:0)
每个笔记本电脑都在不同的计算机上。您需要使用drive.mount(“ drive”)将其保存在Google驱动器上。然后,您可以在不同的笔记本电脑上共享模型。