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实现PS图像明亮度调整效果示例
Jan 23 Python
IntelliJ IDEA安装运行python插件方法
Dec 10 Python
WxPython建立批量录入框窗口
Feb 27 Python
Python小白必备的8个最常用的内置函数(推荐)
Apr 03 Python
PyTorch搭建多项式回归模型(三)
May 22 Python
PyQt 实现使窗口中的元素跟随窗口大小的变化而变化
Jun 18 Python
Python跳出多重循环的方法示例
Jul 03 Python
python错误调试及单元文档测试过程解析
Dec 19 Python
Python中if有多个条件处理方法
Feb 26 Python
python实现飞船游戏的纵向移动
Apr 24 Python
python数据分析之用sklearn预测糖尿病
Apr 22 Python
Python matplotlib安装以及实现简单曲线的绘制
Apr 26 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
强烈推荐:php.ini中文版(2)
2006/10/09 PHP
Javascript 学习书 推荐
2009/06/13 Javascript
JS页面延迟执行一些方法(整理)
2013/11/11 Javascript
手机开发必备技巧:javascript及CSS功能代码分享
2015/05/25 Javascript
轻松实现javascript数据双向绑定
2015/11/11 Javascript
基于javascript实现最简单的选项卡切换效果
2016/05/16 Javascript
jQuery层次选择器用法示例
2016/09/09 Javascript
AngularJs基于角色的前端访问控制的实现
2016/11/07 Javascript
基于jquery二维码生成插件qrcode
2017/01/07 Javascript
vue2.0开发入门笔记之.vue文件的生成和使用
2017/09/19 Javascript
浅析为什么a="abc" 不等于 a=new String("abc")
2017/10/25 Javascript
Bootstrap3.3.7导航栏下拉菜单鼠标滑过展开效果
2017/10/31 Javascript
获取本机IP地址的实例(JavaScript / Node.js)
2017/11/24 Javascript
webpack实用小功能介绍
2018/01/02 Javascript
vue.js自定义组件directives的实例代码
2018/11/09 Javascript
Vue.js中的高级面试题及答案
2020/01/13 Javascript
javascript实现前端成语点击验证优化
2020/06/24 Javascript
微信小程序实现点击页面出现文字
2020/09/21 Javascript
python判断端口是否打开的实现代码
2013/02/10 Python
python基础教程之popen函数操作其它程序的输入和输出示例
2014/02/10 Python
解决Python对齐文本字符串问题
2019/08/28 Python
tensorflow之变量初始化(tf.Variable)使用详解
2020/02/06 Python
PyQt5实现画布小程序
2020/05/30 Python
python读写数据读写csv文件(pandas用法)
2020/12/14 Python
如何利用Python matplotlib绘制雷达图
2020/12/21 Python
CSS3 实现弹幕的示例代码
2017/08/07 HTML / CSS
HTML5 MiranaVideo播放器 (代码开源)
2010/06/11 HTML / CSS
美术专业学生个人自我评价
2013/09/19 职场文书
车间班组长岗位职责
2013/11/13 职场文书
教育学专业实习生的自我鉴定
2013/11/26 职场文书
保护环境倡议书
2014/04/14 职场文书
大学学生会竞选演讲稿
2014/04/25 职场文书
幼儿园感恩节活动方案
2014/10/06 职场文书
民事二审代理词
2015/05/25 职场文书
Python图片验证码降噪和8邻域降噪
2021/08/30 Python
Win11跳过联网界面创建本地管理账户的3种方法
2022/04/20 数码科技