raj*_*dav 2 keras tensorflow google-colaboratory
我正在使用 Google Colaboratory 通过我上传到的数据集来训练我的模型colab.research.google.com。完成训练过程后，我想在本地 PC 中使用训练过的参数，并在本地使用训练过的值创建相同的模型。我应该遵循什么程序？
您需要通过以下方式将您的笔记本连接到您的谷歌驱动器：
from google.colab import drive
drive.mount('/content/gdrive')
然后在训练后，将训练好的模型保存为：
model.save("/content/gdrive/My Drive/name_of_your_model.h5")
打开你的谷歌驱动器，下载保存的文件 name_of_your_model.h5。然后在本地电脑上加载模型
from keras.models import load_model
model = load_model(<path to your model file on local machine>)
现在您在本地 PC 上拥有经过训练的模型
| 归档时间： |
|
| 查看次数： |
3130 次 |
| 最近记录： |