【发布时间】:2020-06-03 03:26:07
【问题描述】:
我在 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版本
否则,我是否必须在训练时将模型保存到本地驱动器的特定路径中?
【问题讨论】:
-
是的,我遵循如下,>from google.colab import drive >drive.mount('/content/drive') '**
-
保存和加载模型的时间差是多少。这是因为您正在保存模型 colab 环境,而 colab 环境在运行时终止时会删除文件。
-
@PSKP 昨天早上 6:30 开始营业,累到早上 7:30 营业。我怀疑这可能是原因。你能告诉我在这种情况下如何将深度学习模型保存到本地驱动器
-
我添加了可以解决您问题的答案。看看吧。