tensorflow 查看模型中的参数列表及对应的值

• tensorflow的checkpoint文件存储模型使用的变量对象，其是一个二进制文件，将变量名映射到对用的tensor值，因此可通过该文件查看参数对应的值

#使用NewCheckpointReader来读取ckpt里的变量
from tensorflow.python import pywrap_tensorflow
# 以下XXX为模型保存chechpoint文件的相对路径，如"model/bestParMTCR_gateBehavior.ckpt"
checkpoint_path = "XXX"  
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) #tf.train.NewCheckpointReader
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
    print("tensor_name: ", key)      # 参数名
    print(reader.get_tensor(key))    # 参数值

---

• 以上版本仅支持tf1, 在tf2环境下可以使用以下方式进行查看：

import os
import tensorflow as tf
import tensorflow.compat.v1 as tfv1
from tensorflow.python import pywrap_tensorflow
path = "ckpt file path"
reader = tfv1.train.NewCheckpointReader(path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
    print("tensor_name: ", key)      # 参数名
    print(reader.get_tensor(key))    # 参数值