时间戳

首页 > 解决方案 > 我应该如何将经过训练的模型从 google colab 部署到我的计算机

tensorflow - 我应该如何将经过训练的模型从 google colab 部署到我的计算机

问题描述

我正在使用 Google Colaboratory 通过我上传到的数据集来训练我的模型colab.research.google.com。完成训练过程后，我想在本地 PC 中使用训练后的参数，并在本地创建具有训练值的相同模型。我应该遵循什么程序？

标签： tensorflowkerasgoogle-colaboratory

解决方案

您需要通过以下方式将笔记本连接到您的 google 驱动器：

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

然后在训练后，将训练好的模型保存为：

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

打开你的谷歌驱动器，下载保存的文件名_of_your_model.h5。然后在本地电脑上加载模型

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

现在你在本地电脑上拥有了训练好的模型

推荐阅读