如何保存在谷歌datalab笔记本中训练的张量流模型进行离线预测?

问题描述 投票:2回答:1

我正在使用Google Cloud Datalab笔记本来训练我的张量流模型。我想保存训练有素的离线预测模型。但是,我对如何保存模型毫无头绪。我应该使用任何张量流模型保存方法还是有任何datalab / google云存储特定方法这样做?在这方面的任何帮助都非常感谢。

tensorflow google-cloud-platform google-cloud-datalab
1个回答
2
投票

您可以使用任何张量流模型保存方法,但我建议您将其保存到Google云端存储分区而不是本地磁盘。大多数tensorflow方法使用gs://前缀接受Google云端存储路径代替文件名。

我建议使用SavedModelBuilder,因为它目前是最便携的。这里有一个例子:https://github.com/GoogleCloudPlatform/cloudml-samples/blob/master/flowers/trainer/model.py#L393

© www.soinside.com 2019 - 2024. All rights reserved.