Tensorflow 如何从checkpoint文件中加载变量名和变量值

假设你已经经过上千次的迭代，并且得到了以下模型：

则从这些checkpoint文件中加载变量名和变量值代码如下：

mnist

下面将给出一个基于卷积神经网络的手写数字识别样例：

补充：查看tensorflow产生的checkpoint文件内容的方法

tensorflow在保存权重模型时多使用tf.train.saver().save 函数进行权重保存，保存的ckpt文件无法直接打开，但tensorflow提供了相关函数 tf.train.newcheckpointreader 可以对ckpt文件进行权重查看。

其中‘modelckpt’是存放.ckpt文件的文件夹，”fc_nn_model”是文件名，如下图所示。

 

var_to_shape_map是一个字典，其中的键值是变量名，对应的值是该变量的形状，如

想要查看某变量值时，需要调用get_tensor函数，即输入以下代码：

以上为个人经验，希望能给大家一个参考，也希望大家多多支持www.887551.com。