使用以下命令:
saved_model_cli show --all --dir output_pb/
输出如下:
saved_model_cli show --all --dir output_pb/1648525103/
2022-03-29 16:59:55.132993: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2022-03-29 16:59:55,178: WARNING tensorflow 140348275545920 Deprecation warnings have been disabled. Set TF_ENABLE_DEPRECATION_WARNINGS=1 to re-enable them.
tensorflow version: 1.15.4, git_version: v1.15.4-788-gef094748fe, compiler_version: 5.3.1 20160406 (Red Hat 5.3.1-6), cxx11_abi_flag: 0, monolithic_build: 0
MetaGraphDef with tag-set: 'serve' contains the following SignatureDefs:
signature_def['serving_default']:
The given SavedModel SignatureDef contains the following input(s):
inputs['examples'] tensor_info:
dtype: DT_STRING
shape: (-1)
name: input_example_tensor:0
The given SavedModel SignatureDef contains the following output(s):
outputs['predictions'] tensor_info:
dtype: DT_INT64
shape: (-1, 256)
name: loss/ArgMax:0
Method name is: tensorflow/serving/predict
|