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开发编码规范
Sep 08 Python
跟老齐学Python之大话题小函数(1)
Oct 10 Python
Python中比较特别的除法运算和幂运算介绍
Apr 05 Python
Python实现短网址ShortUrl的Hash运算实例讲解
Aug 10 Python
Python的自动化部署模块Fabric的安装及使用指南
Jan 19 Python
浅谈python的dataframe与series的创建方法
Nov 12 Python
Python操作mongodb数据库的方法详解
Dec 08 Python
对python3.4 字符串转16进制的实例详解
Jun 12 Python
Python常用模块logging——日志输出功能(示例代码)
Nov 20 Python
Win10里python3创建虚拟环境的步骤
Jan 31 Python
python读取图片的几种方式及图像宽和高的存储顺序
Feb 11 Python
python实现在线翻译
Jun 18 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 动态多文件上传
2009/01/18 PHP
PHP易混淆函数的区别及用法汇总
2014/11/22 PHP
javascript 极速 隐藏/显示万行表格列只需 60毫秒
2009/03/28 Javascript
用JQuery 判断某个属性是否存在hasAttr的解决方法
2013/04/26 Javascript
Js中获取frames中的元素示例代码
2013/07/30 Javascript
Jquery弹出层插件ThickBox的使用方法
2014/12/09 Javascript
jquery制作LED 时钟特效
2015/02/01 Javascript
详解js图片轮播效果实现原理
2015/12/17 Javascript
JS 动态判断PC和手机浏览器实现代码
2016/09/21 Javascript
JS实现的表头列头固定页面功能示例
2017/01/10 Javascript
原生javascript移动端滑动banner效果
2017/03/10 Javascript
原生JS实现隐藏显示图片 JS实现点击切换图片效果
2021/01/27 Javascript
谈谈JS中的!!
2017/12/07 Javascript
jquery登录的异步验证操作示例
2019/05/09 jQuery
解决vue自定义全局消息框组件问题
2019/11/22 Javascript
详解JavaScript修改注册表的方法
2020/01/05 Javascript
Pyhton中防止SQL注入的方法
2015/02/05 Python
在Python中使用列表生成式的教程
2015/04/27 Python
解决python3在anaconda下安装caffe失败的问题
2017/06/15 Python
Python3.6 Schedule模块定时任务(实例讲解)
2017/11/09 Python
python 读取摄像头数据并保存的实例
2018/08/03 Python
详解python中Numpy的属性与创建矩阵
2018/09/10 Python
postman传递当前时间戳实例详解
2019/09/14 Python
Python实现将蓝底照片转化为白底照片功能完整实例
2019/12/13 Python
Silk Therapeutics官网:清洁、抗衰老护肤品
2020/08/12 全球购物
平面设计师的工作职责
2013/11/21 职场文书
中药专业自荐信范文
2014/03/18 职场文书
社区居务公开实施方案
2014/03/27 职场文书
股权投资意向书
2014/04/01 职场文书
安全生产目标责任书
2014/04/14 职场文书
党员个人对照检查材料思想汇报
2014/09/16 职场文书
民事诉讼答辩状范文
2015/05/21 职场文书
pycharm代码删除恢复的方法
2021/06/26 Python
ObjectMapper 如何忽略字段大小写
2021/06/29 Java/Android
Nginx性能优化之Gzip压缩设置详解(最大程度提高页面打开速度)
2022/02/12 Servers
Oracle 11g数据库使用expdp每周进行数据备份并上传到备份服务器
2022/06/28 Oracle