我应该如何将训练有素的模型从Google Colab部署到我的计算机上

时间:2019-02-17 12:47:00

标签: tensorflow keras google-colaboratory

我正在使用Google合作实验室通过我上传到colab.research.google.com的数据集训练模型。完成培训过程后,我想在本地PC中使用经过培训的参数,并在本地创建具有经过培训的值的相同模型。我应该遵循什么程序?

1 个答案:

答案 0 :(得分:2)

您需要通过以下方式将笔记本连接到Google驱动器:

from google.colab import drive
drive.mount('/content/gdrive')

然后在训练后,将训练后的模型另存为:

model.save("/content/gdrive/My Drive/name_of_your_model.h5")

打开您的Google驱动器,下载已保存的文件名name_of_your_model.h5。然后在您的本地PC上加载模型

from keras.models import load_model
model = load_model(<path to your model file on local machine>)

现在您已经在本地PC上拥有训练有素的模型