tensorflow持久化数据格式(2)

所有变量的值都存在下图两个变量中。
文件解释
查看这两个文件的内容:

# 查看变量信息

# 读取checkpoint文件,.data,.index可以省略
reader = tf.train.NewCheckpointReader(r'E:\ZC\procedure\tensorflow_gpu\model\model.ckpt')
# 获取变量名,变量的维度,变量的值
global_variable = reader.get_variable_to_shape_map()
for variable_name in global_variable:
    # 获取变量的取值
    print(+variable_name, global_variable[variable_name], reader.get_tensor(variable_name))

结果
checkpoint文件是tf.train.Saver类自动生成自动维护的,名称也是固定的。
CheckpointState类型的定义:
message ChceckpointState
{
string model_checkpoint_path=1;//最新的模型文件的文件名
repeated string all_model_checkpoint_paths=2;//所有模型文件的文件名

}
在这里插入图片描述

发布了49 篇原创文章 · 获赞 18 · 访问量 1408

猜你喜欢

转载自blog.csdn.net/qq_44099721/article/details/103817692