- tensorflow的checkpoint文件存储模型使用的变量对象,其是一个二进制文件,将变量名映射到对用的tensor值,因此可通过该文件查看参数对应的值
from tensorflow.python import pywrap_tensorflow
checkpoint_path = "XXX"
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
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))
- 以上版本仅支持tf1, 在tf2环境下可以使用以下方式进行查看:
import os
import tensorflow as tf
import tensorflow.compat.v1 as tfv1
from tensorflow.python import pywrap_tensorflow
path = "ckpt file path"
reader = tfv1.train.NewCheckpointReader(path)
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))