keras - 谷歌合作实验室，Keras : Save model in HDF5 file format and download it to Laptop

Question

我正在使用 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 文件加载回 Google Collab
5. 在 Google collab 中恢复培训

Answer 1

有点晚了，但是为了即将到来的开发人员，我会尝试解决这个问题。

该过程与本地计算机中的过程相同，只有两点不同:

要从 Google Collab 下载模型:

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

要将模型上传到 Google Collab:

from google.colab import files
files.upload()

您可以检查此笔记本以获取更多 I/O 选项:https://colab.research.google.com/notebooks/io.ipynb

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