导出具有输入或输出张量名称的Tensorflow的seq2seq模型

时间:2019-02-14 12:03:57

标签: tensorflow export tensorflow-serving seq2seq

我正在使用官方的seq2seq库来生成模型。生成的模型文件是(.data.meta.index.pbtxtcheckpoint

由于文件不是SavedModel格式,因此我无法将其用于Tensorflow服务。

在探索之后,我得到了两个想法:

  1. 使用生成的文件来生成frozen(.pb)文件。但是,我找不到--output_node_names

  2. 所期望的freeze_graph.py
  3. estimator.export_savedmodel内添加train.py,以使用this自动以预期格式生成图形。但是,我找不到names of input tensors

  4. 所期望的serving_input_receiver_fn()

对于master分支中的给定代码,我想知道inputoutput张量的名称。这样,我可以执行以上两种方法之一。

我如何找到它们?在这两种情况下,都需要修改哪些文件或代码部分?

0 个答案:

没有答案