• tensorflow的checkpoint文件存储模型使用的变量对象,其是一个二进制文件,将变量名映射到对用的tensor值,因此可通过该文件查看参数对应的值
#使用NewCheckpointReader来读取ckpt里的变量
from tensorflow.python import pywrap_tensorflow
# 以下XXX为模型保存chechpoint文件的相对路径,如"model/bestParMTCR_gateBehavior.ckpt"
checkpoint_path = "XXX"  
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))    # 参数值

  • 以上版本仅支持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()
# Print tensor name and values
for key in var_to_shape_map:
    print("tensor_name: ", key)      # 参数名
    print(reader.get_tensor(key))    # 参数值
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐