所有变量的值都存在下图两个变量中。
查看这两个文件的内容:
# 查看变量信息
# 读取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;//所有模型文件的文件名
}