TensorFlow的convert_variables_to_constants函数
来源:互联网 发布:单片机音乐程序 编辑:程序博客网 时间:2024/06/05 19:54
前面介绍了通过使用tf.train.Saver函数来保存TensorFlow程序的参数,但是,在使用tf.train.Saver函数保存模型文件的时候,是保存所有的参数信息,而有些时候我们并不需要所有的参数信息。我们只需要知道神经网络的输入层经过前向传播计算得到输出层即可,所以在保存的时候,我们也不需要保存所有的参数,以及变量的初始化、模型保存等辅助节点信息与迁移学习类似。之前使用tf.train.Saver函数保存模型文件的时候会产生多个文件,它将变量的取值和计算图结构分成了不同的文件存储。TensorFlow提供了另一种保存模型文件的方法,将计算图保存在一个文件中。
1、模型文件的保存
convert_variables_to_constants函数,会将计算图中的变量取值以常量的形式保存。在保存模型文件的时候,我们只是导出了GraphDef部分,GraphDef保存了从输入层到输出层的计算过程。在保存的时候,通过convert_variables_to_constants函数来指定保存的节点名称而不是张量的名称,“add:0”是张量的名称而"add"表示的是节点的名称。
2、模型文件的读取
在读取模型文件获取变量的值的时候,我们需要指定的是张量的名称而不是节点的名称,这两个地方需要特别注意一下。在读取模型文件的时候,可能会遇到一个错误tensorflow.python.framework.errors_impl.NotFoundError: NewRandomAccessFile failed to Create/Open: ./model.pd,打开模型文件的时候报错,所以一般都是模型文件的名称有问题,要注意统一。在这个地方,我遇到了一个很奇葩的问题,我读取模型文件的名称和保存的文件名称是一样,但是还是报错,后面我将保存模型文件的名称复制到读取模型文件的名称那里,就解决了这个问题。
阅读全文
0 0
- TensorFlow的convert_variables_to_constants函数
- TensorFlow的convert_variables_to_constants函数
- tensorflow将训练好的模型freeze,即将权重固化到图里面,并使用该模型进行预测(tf.graph_util.convert_variables_to_constants函数)
- TensorFlow的reduce_sum()函数
- tensorflow的函数
- TensorFlow的variables_to_restore函数
- Tensorflow | 函数名字的更新
- tensorflow 函数接口的理解
- TensorFlow 里的线性函数
- Tensorflow里的reshape函数
- tensorflow的一些基本函数
- Tensorflow的一些常见函数
- Tensorflow 学习与复习 Epoch_1 # tensorflow的一些函数方法
- 【Tensorflow】Tensorflow一些数据IO、图的运行相关函数
- tensorflow的tf.equal()函数的使用方法
- tensorflow的tf.reduce_mean()函数的使用方法
- TensorFlow从图像提取区域的函数
- tensorflow 一些常用函数的功能
- 自动驾驶汽车会让道路更安全—你敢坐吗?---凯利讯半导体
- [Play 2.3.8 Error with [entity.manager.HitInfo] It has not been enhanced but it's superClass [class
- C语言小知识(2)
- Ubuntu搭建web服务器系列之JDK(JDK+Tomcat+MySQL+Nginx+Redis+NodeJS)
- 手动在电脑桌面制作关机/重启图标并添加到任务栏
- TensorFlow的convert_variables_to_constants函数
- 设计模式:建造者模式
- spark createDirectStream保存kafka offset(JAVA实现)
- suseLinux下安装oracle12c
- quartz,spring-quartz,spring-task,spring-schedule
- Bootstrap表格插件--bootstrap-table
- VMware虚拟机 Ubuntu 16.04 安装 VMware Tools
- Glide使用特点
- **AngularJs学习笔记1**