Google colaboratory,Keras:以HDF5文件格式保存模型并将其下载到笔记本电脑

时间:2018-05-21 10:50:25

标签: keras hdf5 h5py google-colaboratory

我正在使用GPU在Google Collab中训练小型RNN模型。我通常以HDF5文件格式保存我的模型和权重。

在本地机器(笔记本电脑)中,我按照以下程序进行操作

sudo pip install h5py
model.fit(....)
model.save('model1.h5')

我加载回训练的模型以使用

进行预测
from keras.models import load_model
model = load_model('model1.h5')

我现在想要

  1. 在Google Collab中保存模型,格式与上面类似
  2. 将.h5文件下载到本地计算机(PC)
  3. 在PC中进行预测并在PC中进行训练
  4. 保存经过培训的模型,将.h5文件加载回谷歌协作
  5. 恢复Google协作培训

1 个答案:

答案 0 :(得分:1)

有点晚了,但是,为迎接即将到来的开发人员,我将尝试解决问题:

此过程与本地计算机中的过程相同,只是有两个区别:

要从Google Collab下载模型:

from google import files
files.download("model1.h5")

要将模型上传到Google Collab:

from google import files
files.upload("model1.h5")

您可以在此笔记本上查看更多的输入/输出选项:https://colab.research.google.com/notebooks/io.ipynb

所有其他步骤的执行方式与您在本地计算机上执行的方式相同。 希望对您有所帮助。