tensorflow小技巧之查看保存模型参数的 name 和 value

tensorflow小技巧之查看保存模型参数的 name 和 value

在将tensorflow训练好的模型保存之后，可能会遇见需要查看其中的权重和偏置的名称以及具体的值的大小，可以使用tensorflow.python 中的 pywrap_tensorflow库进行查看
具体程序如下：

import os
from tensorflow.python import pywrap_tensorflow
num_epoch = 199
model_path = r'checkpoint\Autoencoder_128_Pixel_Grad\Autoencoder.model-' + str(num_epoch)
model_path = os.path.join(os.getcwd(), model_path)
rule_num_epoch = 49
rule_model_path = './checkpoint/UNet_Fusion_Rule_128_MSE/RuleModel.model-' + str(rule_num_epoch)
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(rule_model_path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
    print("name: ", key)
    print(reader.get_tensor(key))

模型保存的形式如上图所示。
查看结果则如下图所示。

因为值无法完全显示，所以只展示name。