我正在使用Google Cloud Datalab笔记本来训练我的张量流模型。我想保存训练有素的离线预测模型。但是,我对如何保存模型毫无头绪。我应该使用任何张量流模型保存方法还是有任何datalab / google云存储特定方法这样做?在这方面的任何帮助都非常感谢。
您可以使用任何张量流模型保存方法,但我建议您将其保存到Google云端存储分区而不是本地磁盘。大多数tensorflow方法使用gs://前缀接受Google云端存储路径代替文件名。
我建议使用SavedModelBuilder,因为它目前是最便携的。这里有一个例子:https://github.com/GoogleCloudPlatform/cloudml-samples/blob/master/flowers/trainer/model.py#L393