无法部署Cloud ML模型

时间:2016-09-30 00:30:14

标签: google-cloud-ml

为什么我尝试将训练过的模型部署到Google Cloud ML,我收到以下错误:

Create Version failed.Model validation failed: Model metagraph does not have inputs collection.

这是什么意思以及如何解决这个问题?

1 个答案:

答案 0 :(得分:2)

部署在CloudML上的Tensorflow模型没有名为“inputs”的集合。此集合应为您的图表命名所有输入张量。同样,需要一个名为“outputs”的集合来命名图形的输出张量。假设您的图形有两个输入张量x和y,以及一个输出张量分数,可以按如下方式完成:

tf.add_to_collection(“inputs”, json.dumps({“x” : x.name, “y”: y.name}))
tf.add_to_collection(“outputs”, json.dumps({“scores”: scores.name}))

这里“x”,“y”和“得分”成为实际张量名称的别名(x.name,y.name和scores.name)