获取savemodel的输入输出节点
saved_model_cli show --dir savemodels --all
结果:
MetaGraphDef with tag-set: ‘serve’ contains the following SignatureDefs:
signature_def[‘translation_signature’]:
The given SavedModel SignatureDef contains the following input(s):
inputs[‘input_word_ids’] tensor_info:
dtype: DT_INT64
shape: (-1, -1)
name: input_wids_1:0
The given SavedModel SignatureDef contains the following output(s):
outputs[‘output_trans_result’] tensor_info:
dtype: DT_INT32
shape: (-1, 4, -1)
name: NmtModel_1/strided_slice_9:0
Method name is: tensorflow/serving/predict