我正在使用官方的seq2seq库来生成模型。生成的模型文件是(.data
,.meta
,.index
,.pbtxt
,checkpoint
)
由于文件不是SavedModel
格式,因此我无法将其用于Tensorflow服务。
在探索之后,我得到了两个想法:
使用生成的文件来生成frozen(.pb)
文件。但是,我找不到--output_node_names
freeze_graph.py
在estimator.export_savedmodel
内添加train.py
,以使用this自动以预期格式生成图形。但是,我找不到names of input tensors
serving_input_receiver_fn()
对于master分支中的给定代码,我想知道input
和output
张量的名称。这样,我可以执行以上两种方法之一。
我如何找到它们?在这两种情况下,都需要修改哪些文件或代码部分?