我正在使用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')
我现在想要
答案 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
所有其他步骤的执行方式与您在本地计算机上执行的方式相同。 希望对您有所帮助。