TensorFlow 读取 CKPT 模型变量名与数值的方法

月亮慢半拍

发布时间：2026-10-13| 4人浏览过

原创

本文介绍如何利用 TensorFlow 的 pywrap_tensorflow 模块读取保存的 CKPT 模型文件。通过 NewCheckpointReader 获取变量形状映射，并遍历打印变量名称及具体张量值，适用于模型训练后的调试与验证场景。

TensorFlow 读取 CKPT 模型变量名与数值的方法

在模型训练完成后，有时需要检查保存的 CKPT 文件中包含的具体变量数值，以验证训练效果或进行调试。

标准的 CKPT 模型文件通常由以下三个部分组成：

数据文件，格式为 xxxxx.ckpt.data-00000-of-00001

索引文件，格式为 xxxxx.ckpt.index

元数据文件，格式为 xxxxx.ckpt.meta


import os
from tensorflow.python import pywrap_tensorflow

checkpoint_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 模型中保存的变量名称及数值的具体实现方式，供读者参考与使用。