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;//所有模型文件的文件名

}
在这里插入图片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章