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