google-cloud-ml - 无法部署 Cloud ML 模型

Question

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

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

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

Answer 1

部署在 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”和“scores”成为实际张量名称的别名(x.name，y.name和scores.name)