欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-06-23 09:43:29
假设你已经经过上千次的迭代,并且得到了以下模型:则从这些checkpoint文件中加载变量名和变量值代码如下:mnist下面将给出一个基于卷积神经网络的手写数字识别样例:补充:查看tensorflow...

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

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

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

mnist

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

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

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

其中‘modelckpt'是存放.ckpt文件的文件夹,"fc_nn_model"是文件名,如下图所示。

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

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

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

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。