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装饰器 免去调用父类构造函数的麻烦
May 18 Python
Python标准库inspect的具体使用方法
Dec 06 Python
python+opencv轮廓检测代码解析
Jan 05 Python
Python中的pack和unpack的使用
Mar 12 Python
Centos7 Python3下安装scrapy的详细步骤
Mar 15 Python
Python3+django2.0+apache2+ubuntu14部署网站上线的方法
Jul 07 Python
Python给定一个句子倒序输出单词以及字母的方法
Dec 20 Python
Django RBAC权限管理设计过程详解
Aug 06 Python
python中return的返回和执行实例
Dec 24 Python
Python调用shell cmd方法代码示例解析
Jun 18 Python
python中的split、rsplit、splitlines用法说明
Oct 23 Python
在python3.9下如何安装scrapy的方法
Feb 03 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中Date获取时间不正确怎么办
2008/06/05 PHP
php将会员数据导入到ucenter的代码
2010/07/18 PHP
第4章 数据处理-php数组的处理-郑阿奇
2011/07/04 PHP
php实例分享之mysql数据备份
2014/05/19 PHP
php中file_get_content 和curl以及fopen 效率分析
2014/09/19 PHP
PHP高精确度运算BC函数库实例详解
2017/08/15 PHP
php 删除一维数组中某一个值元素的操作方法
2018/02/01 PHP
Laravel 类和接口注入相关的代码
2019/10/15 PHP
laravel邮件发送的实现代码示例
2020/01/31 PHP
jQuery动态创建html元素的常用方法汇总
2014/09/05 Javascript
javascript中typeof操作符和constucor属性检测
2015/02/26 Javascript
JavaScript中的setMilliseconds()方法使用详解
2015/06/11 Javascript
理解javascript中的with关键字
2016/02/15 Javascript
JavaScript必知必会(九)function 说起 闭包问题
2016/06/08 Javascript
js改变html的原有内容实现方法
2016/10/05 Javascript
d3.js入门教程之数据绑定详解
2017/04/28 Javascript
微信小程序的日期选择器的实例详解
2017/09/29 Javascript
BootStrap table实现表格行拖拽效果
2018/12/01 Javascript
微信小程序使用map组件实现获取定位城市天气或者指定城市天气数据功能
2019/01/22 Javascript
详解element-ui动态限定的日期范围选择器代码片段
2020/07/03 Javascript
[01:35:53]完美世界DOTA2联赛PWL S3 Magma vs GXR 第二场 12.13
2020/12/17 DOTA
python创建线程示例
2014/05/06 Python
在Python的Django框架中实现Hacker News的一些功能
2015/04/17 Python
python使用in操作符时元组和数组的区别分析
2015/05/19 Python
Python实现购物程序思路及代码
2017/07/24 Python
用Anaconda安装本地python包的方法及路径问题(图文)
2019/07/16 Python
浅谈pytorch grad_fn以及权重梯度不更新的问题
2019/08/20 Python
Python 远程开关机的方法
2020/11/18 Python
Lululemon加拿大官网:加拿大知名体育服装零售商
2019/04/12 全球购物
西班牙最大的婴儿用品网上商店:Bebitus
2019/05/30 全球购物
Woods官网:加拿大最古老、最受尊敬的户外品牌之一
2020/09/12 全球购物
大学毕业登记表自我鉴定
2013/10/09 职场文书
个人银行贷款担保书
2014/04/01 职场文书
简单通用的简历自我评价
2014/09/21 职场文书
解除劳动关系协议书2篇
2014/11/28 职场文书
2014预防青少年违法犯罪工作总结
2014/12/10 职场文书