keras - Google COLAB 免费版本保存 Keras 训练模型

Question

我在Google Colab免费版本中保存了keras训练的模型

 model.save("my_model.h5")

我尝试使用以下方法检索模型

from keras.models import load_model
model = load_model('my_model.h5')

但它抛出错误

OSError: Unable to open file (unable to open file: name = 'my_model.h5', errno = 2, error message = 'No such file or directory', flags = 0, o_flags = 0)

我可以从免费的 Google Colab 版本中检索保存的模型吗，您对此有什么帮助吗 我在stackoverflow中检查了类似的问题，我认为这些答案属于colab pro版本

否则，在训练时我是否必须将模型保存到本地驱动器的特定路径中？

Answer 1

问题是什么

您正在运行时存储模型，而不是存储在 Google 云端硬盘中。运行12小时后自动删除数据。所以我们必须将模型保存在Google Drive中。

如何存储到 Google 云端硬盘

首先连接到 Google 云端硬盘

from google.colab import drive
drive.mount('/content/drive')

现在您将在左侧找到文件资源管理器，其中有drive目录。当您进入该目录时，它将带您进入谷歌驱动器。

假设我想将数据放入驱动器我的驱动器，然后

from keras.models import load_model

MODEL_PATH = './drive/My Drive/model.h5'

# Now save model in drive
model.save(MODEL_PATH)

# Load Model
model = load_model(MODEL_PATH)

当您打开驱动器时，您将在驱动器中找到文件model.h5。