我应该如何将经过训练的模型从 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>)
现在你在本地电脑上有了训练好的模型
我正在使用 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>)
现在你在本地电脑上有了训练好的模型