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

Question

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

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)

但这会引发错误

.transformEffect(.init(scaleX: 0.7, y: 0.7))

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

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

Answer 1

什么是问题

您是在运行时中存储模型，而不是在Google驱动器中存储模型。 12小时后，运行时会自动删除数据。因此，我们必须将模型保存在Google驱动器中。

如何存储到Google云端硬盘

首先连接到Google云端硬盘

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

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

然后我想将数据放入驱动器My 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。