tensorflow 加载部分变量的实例讲解


Posted in Python onJuly 27, 2018

tensorflow模型保存为saver = tf.train.Saver()函数,saver.save()保存模型,代码如下:

import tensorflow as tf
 
v1= tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="v1")
v2= tf.Variable(tf.zeros([200]), name="v2")
saver = tf.train.Saver()
with tf.Session() as sess:
 init_op = tf.global_variables_initializer()
 sess.run(init_op)
 saver.save(sess,"checkpoint/model_test",global_step=1)

当我们保存模型后,我们可以通过saver.restore()来加载模型,初始化变量:

import tensorflow as tf
 
v1= tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="v1")
v2= tf.Variable(tf.zeros([200]), name="v2")
saver = tf.train.Saver()
with tf.Session() as sess:
 # init_op = tf.global_variables_initializer()
 # sess.run(init_op)
 saver.restore(sess, "checkpoint/model_test-1")
 # saver.save(sess,"checkpoint/model_test",global_step=1)

神经网络训练时,有时候我们需要从预训练的模型中加载部分参数,初始化当前模型,例如加入CNN有6层,我们需要从已有的模型初始化CNN前5层参数.这可以通过saver.restore()实现.

之前我们已经介绍可以通过tf.train.Saver()的保存部分变量的方法,即需要保存的变量列表,同样的,在变量初始化的时候,我们可以对需要单独初始化的变量分别定义一个tf.train.Saver()函数,这样就可以单独对该部分变量初始化,例如下面代码,saver1用于初始化变量v1,saver2用于初始化变量v2,v3:

import tensorflow as tf
 
v1= tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="v1")
v2= tf.Variable(tf.zeros([200]), name="v2")
v3= tf.Variable(tf.zeros([100]), name="v3")
#saver = tf.train.Saver()
saver1 = tf.train.Saver([v1])
saver2 = tf.train.Saver([v2]+[v3])
with tf.Session() as sess:
 # init_op = tf.global_variables_initializer()
 # sess.run(init_op)
 saver1.restore(sess, "checkpoint/model_test-1")
 saver2.restore(sess, "checkpoint/model_test-1")
 # saver.save(sess,"checkpoint/model_test",global_step=1)

以上这篇tensorflow 加载部分变量的实例讲解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python实现的二维码生成小软件
Jul 11 Python
python urllib爬取百度云连接的实例代码
Jun 19 Python
关于Python中空格字符串处理的技巧总结
Aug 10 Python
对Python 内建函数和保留字详解
Oct 15 Python
python分批定量读取文件内容,输出到不同文件中的方法
Dec 08 Python
如何用Python破解wifi密码过程详解
Jul 12 Python
python实现socket+threading处理多连接的方法
Jul 23 Python
Python定时任务工具之APScheduler使用方式
Jul 24 Python
Python 异常的捕获、异常的传递与主动抛出异常操作示例
Sep 23 Python
python基于plotly实现画饼状图代码实例
Dec 16 Python
Python callable内置函数原理解析
Mar 05 Python
Pycharm安装python库的方法
Nov 24 Python
Python基于递归算法求最小公倍数和最大公约数示例
Jul 27 #Python
Python切片操作深入详解
Jul 27 #Python
对Tensorflow中的变量初始化函数详解
Jul 27 #Python
JavaScript中的模拟事件和自定义事件实例分析
Jul 27 #Python
浅谈tensorflow中几个随机函数的用法
Jul 27 #Python
tensorflow 中对数组元素的操作方法
Jul 27 #Python
Python读取excel中的图片完美解决方法
Jul 27 #Python
You might like
超外差式晶体管收音机的组装与统调
2021/03/01 无线电
php带密码功能并下载远程文件保存本地指定目录 修改加强版
2010/05/16 PHP
举例讲解PHP面对对象编程的多态
2015/08/12 PHP
YII2框架中ActiveDataProvider与GridView的配合使用操作示例
2020/03/18 PHP
给网站上的广告“加速”显示的方法
2007/04/08 Javascript
Javascript的匿名函数小结
2009/12/31 Javascript
基于jquery的关于动态创建DOM元素的问题
2010/12/24 Javascript
jquery获得下拉框值的代码
2011/08/13 Javascript
jquery text(),val(),html()方法区别总结
2013/11/04 Javascript
jquery1.9 下检测浏览器类型和版本的方法
2013/12/26 Javascript
jQuery异步验证用户名是否存在示例代码
2014/05/21 Javascript
Jquery on方法绑定事件后执行多次的解决方法
2016/06/02 Javascript
浅谈MVC+EF easyui dataGrid 动态加载分页表格
2016/11/10 Javascript
JS实现的图片预览插件与用法示例【不上传图片】
2016/11/25 Javascript
ES6学习笔记之正则表达式和字符串正则方法分析
2017/04/25 Javascript
js实现日期显示的一些操作(实例讲解)
2017/07/27 Javascript
React-Native做一个文本输入框组件的实现代码
2017/08/10 Javascript
vue 动态修改a标签的样式的方法
2018/01/18 Javascript
JS实现数组的增删改查操作示例
2018/08/29 Javascript
Vue v-for循环之@click点击事件获取元素示例
2019/11/09 Javascript
原生JS实现无缝轮播图片
2020/06/24 Javascript
JS获取当前时间戳方法解析
2020/08/29 Javascript
vue下载二进制流图片操作
2020/10/26 Javascript
Python列表推导式、字典推导式与集合推导式用法实例分析
2018/02/07 Python
python多线程抽象编程模型详解
2019/03/20 Python
python 装饰器功能与用法案例详解
2020/03/06 Python
Python基于Tkinter编写crc校验工具
2020/05/06 Python
HTML5 Canvas实现烟花绽放特效
2016/03/02 HTML / CSS
毕业学生推荐信
2013/12/01 职场文书
公司新员工的演讲稿注意事项
2014/01/01 职场文书
2014年教师党员自我评价范文
2014/09/22 职场文书
酒店办公室主任岗位职责
2015/04/01 职场文书
一年级下册数学教学反思
2016/02/16 职场文书
学生安全责任协议书
2016/03/22 职场文书
李清照的诗词赏析(20首)
2019/08/22 职场文书
pandas 操作 Excel操作总结
2021/03/31 Python