tensorflow convert_variables_to_constants
2024-09-02 00:08:13
在使用tf.train.Saver函数保存模型文件的时候,是保存所有的参数信息,而有些时候我们并不需要所有的参数信息。我们只需要知道神经网络的输入层经过前向传播计算得到输出层即可,所以在保存的时候,我们也不需要保存所有的参数,以及变量的初始化、模型保存等辅助节点信息与迁移学习类似。之前使用tf.train.Saver函数保存模型文件的时候会产生多个文件,它将变量的取值和计算图结构分成了不同的文件存储。TensorFlow提供了另一种保存模型文件的方法,将计算图保存在一个文件中。
tmp_g = tf.get_default_graph().as_graph_def()
tmp_g = tf.graph_util.convert_variables_to_constants(sess, tmp_g, [n.name[:-2] for n in output_tensors])
最新文章
- distribution 中一直在运行 waitfor delay @strdelaytime 语句
- 防SQL注入
- ngCordova
- Java 2D API - 1. 基本概念
- 9 HTML&;JS等前端知识系列之Ajax post请求带有token向Django请求
- 根据不同的屏幕宽度引入不同的css文件
- MVC 区分是哪按键提交FORM
- MeasureSpec介绍
- HTTP协议 流程图
- HTML&;CSS基础学习笔记1-简单网页中有哪些标签?
- 数据可视化之MarkPoint
- window10:jdk 8下载和安装步骤
- 安装和配置bazel
- 部署GitLab代码托管仓库
- html5-绝对路径/相对路径
- tomcat设置默认启动项
- 问题2:css图片、文字居中
- 在学习使用webpack中遇到的错误
- SD从零开始19-20
- 【PHP 】伪静态 - 4. 实际运用
热门文章
- VMware中Bridged、NAT、host-only三种网络连接模式的原理及其区别
- 来看下,C# WebService WSDL自动生成代码,数组参数的BUG。。。ArrayOfString
- CefSharp禁止弹出新窗体,在同一窗口打开链接,或者在新Tab页打开链接,并且支持带type=";POST"; target=";_blank";的链接
- Python活力练习Day2
- 墨者 - X-FORWARDED-FOR注入漏洞实战
- maven的下载、安装及配置
- sonar安装和使用
- python字符串与字典转换
- C语言笔记 05_判断&;循环&;函数
- AES加密原理和AOE工程实践