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

How should I deploy the trained model from google colab to my computer

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

您需要通过以下方式将您的笔记本电脑连接到您的 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。然后在您的本地电脑上加载模型

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

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