Google colaboratory，Keras：以HDF5文件格式保存模型并下载到笔记本电脑

我正在使用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')

我现在想

将模型保存在GoogleCollab中，格式与上述类似

将.h5文件下载到本地计算机（PC）

在PC机上进行预测并在PC机上进行训练

保存经过培训的模型，将.h5文件加载回google collab

恢复谷歌合作培训

---

有点晚了，但是为了未来的开发者，我将尝试解决这个问题

该过程与本地计算机中的过程相同，只是有两个区别：

从Google Collab下载模型：

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

from google.colab import files
files.upload()

要将模型上载到Google Collab:

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

from google.colab import files
files.upload()

您可以查看此笔记本以了解更多i/o选项：

所有其他步骤的执行方式与您在本地计算机中的执行方式相同。

---

希望这能对您有所帮助。

更正

来自google.colab导入文件

另外，files.upload（）不接受任何参数，因此这不起作用。