Tensorflow 如何从checkpoint文件中加载变量名和变量值
程序员文章站
2022-06-23 09:43:29
假设你已经经过上千次的迭代,并且得到了以下模型:则从这些checkpoint文件中加载变量名和变量值代码如下:mnist下面将给出一个基于卷积神经网络的手写数字识别样例:补充:查看tensorflow...
假设你已经经过上千次的迭代,并且得到了以下模型:
则从这些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函数,即输入以下代码:
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。