查看tensorflow ckpt模型的输出节点

查看tensorflow ckpt模型的输出节点

（1）查看你的代码

output_node_names = “score/output”
转自 https://www.jianshu.com/p/06548e3e8f4b

（2）查看TensorFlow checkpoint文件中的变量名和对应值

from tensorflow.python import pywrap_tensorflow
import os
checkpoint_path = os.path.join('./ckpt/best_validation')
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
    print('tensor_name: ', key)

运行结果：
参照 https://blog.csdn.net/helei001/article/details/56489658
（3）用tensorboard查看CKPT图结构
https://blog.csdn.net/weixin_43815222/article/details/108075416