tensorflow查看保存模型的参数
参考博客:
tensorflow 模型存储、查看与载入
tensorflow小技巧之查看保存模型参数的 name 和 value
from tensorflow.python import pywrap_tensorflow
model_path = 'your/path/to/model/model-12370'
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(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)