Tensorflow-从具有固定批次大小1的估算器保存.pb

时间:2018-09-25 15:53:14

标签: python tensorflow tensorflow-serving tensorflow-datasets tensorflow-estimator

我想问问有什么方法可以将pb文件保存为固定的批次大小1,以便通过Estimator进行生产。

我可以使用会话样式,因为它只声明了一个批处理大小为1的会话。

How to store model in `.pb` file with Estimator in TensorFlow?

在这里,有一些示例如何将估算器检查点保存到pb文件中。

我可以保存批量大小固定为1的pb二进制文件,并从Java tensorflow之类的其他平台加载。

我认为有两个问题。

1。保存具有特定图层名称的pb二进制文件,但不知道如何保存为批处理大小1。

2。当我使用estimator时,我无法定义输入名称,因为我仅将tf.data用于estimator(建议至少使用tensorflow)。

例如,我不确定如何使用tf.data在图中定义名称 像:

句子= tf.placeholder([批量大小,无,嵌入大小],名称= “句子输入”

感谢你们的帮助。

0 个答案:

没有答案