我想问问有什么方法可以将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([批量大小,无,嵌入大小],名称= “句子输入” )
感谢你们的帮助。