TensorFlow 读取 CKPT 模型变量名与数值的方法
发布时间：2026-10-13 |
5人浏览过
本文介绍如何利用 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 模型中保存的变量名称及数值的具体实现方式，供读者参考与使用。
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
