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