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写的图片蜘蛛人代码
Aug 27 Python
python采集百度百科的方法
Jun 05 Python
使用tensorflow实现AlexNet
Nov 20 Python
Pandas实现数据类型转换的一些小技巧汇总
May 07 Python
Python实现基于C/S架构的聊天室功能详解
Jul 07 Python
查看python下OpenCV版本的方法
Aug 03 Python
Python3爬楼梯算法示例
Mar 04 Python
如何为Python终端提供持久性历史记录
Sep 03 Python
Python numpy线性代数用法实例解析
Nov 15 Python
Python3变量与基本数据类型用法实例分析
Feb 14 Python
keras load model时出现Missing Layer错误的解决方式
Jun 11 Python
如何卸载python插件
Jul 08 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 清除网页病毒的方法
2008/12/05 PHP
PHP自动识别字符集并完成转码详解
2013/08/02 PHP
CentOS安装php v8js教程
2015/02/26 PHP
PHP实现动态添加XML中数据的方法
2018/03/30 PHP
JavaScript的Cookies
2008/01/16 Javascript
javascript的数据类型、字面量、变量介绍
2012/05/23 Javascript
js事件监听机制(事件捕获)总结
2014/08/08 Javascript
Nodejs学习笔记之测试驱动
2015/04/16 NodeJs
js自定义回调函数
2015/12/13 Javascript
浅谈bootstrap源码分析之tab(选项卡)
2016/06/06 Javascript
Wireshark基本介绍和学习TCP三次握手
2016/08/15 Javascript
Angular1.x自定义指令实例详解
2017/03/01 Javascript
jQuery简易时光轴实现方法示例
2017/03/13 Javascript
jQuery插件FusionCharts实现的MSBar2D图效果示例【附demo源码】
2017/03/24 jQuery
浅谈Node.js轻量级Web框架Express4.x使用指南
2017/05/03 Javascript
jQuery实现表格冻结顶栏效果
2017/08/20 jQuery
Vue实现类似Spring官网图片滑动效果方法
2019/03/01 Javascript
小程序异步问题之多个网络请求依次执行并依次收集请求结果
2019/05/05 Javascript
Python的装饰器模式与面向切面编程详解
2015/06/21 Python
python 查找文件名包含指定字符串的方法
2018/06/05 Python
python实现的爬取电影下载链接功能示例
2019/08/26 Python
django处理select下拉表单实例(从model到前端到post到form)
2020/03/13 Python
Django利用elasticsearch(搜索引擎)实现搜索功能
2020/11/26 Python
HTML5实现简单图片上传所遇到的问题及解决办法
2016/01/20 HTML / CSS
Mavi牛仔裤美国官网:土耳其著名牛仔品牌
2016/09/24 全球购物
美国最大的船只买卖在线市场:Boat Trader
2018/08/04 全球购物
递归计算如下递归函数的值(斐波拉契)
2012/02/04 面试题
优秀信贷员先进事迹
2014/01/31 职场文书
信息专业学生学习的自我评价
2014/02/17 职场文书
新文化运动的基本口号
2014/06/21 职场文书
2014党支部对照检查材料思想汇报
2014/10/05 职场文书
创业项目(超低成本创业项目)
2019/08/16 职场文书
手把手教你制定暑期学习计划,让你度过充实的暑假
2019/08/22 职场文书
SQL基础的查询语句
2021/11/11 MySQL
python turtle绘图命令及案例
2021/11/23 Python
Windows Server 2008配置防火墙策略详解
2022/06/28 Servers