【tensorflow】打印ckpt模型保存下的变量名称及变量值

本文主要是介绍【tensorflow】打印ckpt模型保存下的变量名称及变量值，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

有时候会需要通过从保存下来的ckpt文件来观察其保存下来的训练完成的变量值。

ckpt文件名列表：（一般是三个文件）

• xxxxx.ckpt.data-00000-of-00001
• xxxxx.ckpt.index
• xxxxx.ckpt.meta

import os
from tensorflow.python import pywrap_tensorflowcheckpoint_path = os.path.join("文件夹路径", "xxxxx.ckpt")
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))

这篇关于【tensorflow】打印ckpt模型保存下的变量名称及变量值的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---