问题是关于Tensorflow及其更高的API估算器,更一般地说是关于导出和重用模型。
存在几种导出张量流模型的工具:tf.train.Saver,直接写入和导入图形及其相关权重。
我的兴趣在于tf.estimator.Estimator对象以及我们如何重用它们。我的用例非常简单:我在tensorflow中使用一个简单的模型,并希望在更复杂的一个中使用它的预测。
存在用于服务模型的export_savemodel
方法,但是我需要通过端口在我的模型中进行通信,并且我不确定这是否是训练的最佳选择。
我的问题如下:
export_savemodel
的输出?我猜你可以在技术上使用C ++代码并围绕它们创建包装器?tf.estimator.Estimator
并在Python中重用它? XLA编译器似乎很有趣,但我无法知道我是否可以在Python中使用结果。非常感谢您的阅读!
编辑(我自己的答案):接受的答案提供使用saved_model
API。但是,我认为理解图形和冻结图形的概念是值得的。这个gist link说明了可能在saved_model
api中复制的主要步骤。基本上,你必须:
input_map
重新加载模型时,为tf.import_graph_def
参数提供新输入。