如何在不重新训练模型的情况下更改SavedModel的签名?

时间:2017-03-15 05:11:41

标签: tensorflow tensorflow-serving google-cloud-ml google-cloud-ml-engine

我刚刚完成了对模型的训练,结果发现我导出了一个用于签名问题的服务模型。我如何更新它们?

(一个常见问题是为CloudML Engine设置错误的形状。)

1 个答案:

答案 0 :(得分:3)

不用担心 - 您不需要重新训练模型。也就是说,还有一些工作要做。您将创建一个新的(更正的)服务图,将检查点加载到该图中,然后导出此图。

例如,假设您添加占位符,但未设置形状,即使您打算(例如,在CloudML上运行)。在这种情况下,您的图表可能如下所示:

mapper

要纠正这个问题:

x = tf.placeholder(tf.float32)
y = foo(x)
...