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 str与repr的区别
Mar 23 Python
Python实例之wxpython中Frame使用方法
Jun 09 Python
跟老齐学Python之网站的结构
Oct 24 Python
python实现在目录中查找指定文件的方法
Nov 11 Python
Python读取本地文件并解析网页元素的方法
May 21 Python
python的pip安装以及使用教程
Sep 18 Python
使用Python刷淘宝喵币(低阶入门版)
Oct 30 Python
Python Django框架模板渲染功能示例
Nov 08 Python
Pytest mark使用实例及原理解析
Feb 22 Python
python调用摄像头的示例代码
Sep 28 Python
Python实现灰色关联分析与结果可视化的详细代码
Mar 25 Python
Python FuzzyWuzzy实现模糊匹配
Apr 28 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简单开启gzip压缩方法(zlib.output_compression)
2013/04/13 PHP
浅析十款PHP开发框架的对比
2013/07/05 PHP
php中3种方法删除字符串中间的空格
2014/03/10 PHP
PHP实现的注册,登录及查询用户资料功能API接口示例
2017/06/06 PHP
PDO::inTransaction讲解
2019/01/28 PHP
对采用动态原型方式无法展示继承机制得思考
2009/12/04 Javascript
js 获取服务器控件值的代码
2010/03/05 Javascript
javascript学习基础笔记之DOM对象操作
2011/11/03 Javascript
JS+CSS制作DIV层可(最小化/拖拽/排序)功能实现代码
2013/02/25 Javascript
JS随机生成不重复数据的实例方法
2013/07/17 Javascript
javascript删除一个html元素节点的方法
2014/12/20 Javascript
Jquery日期选择datepicker插件用法实例分析
2015/06/08 Javascript
jQuery实现连续动画效果实例分析
2015/10/09 Javascript
深入理解jquery中的each用法
2016/12/14 Javascript
浅谈struts1 & jquery form 文件异步上传
2017/05/25 jQuery
Vue2.x中的Render函数详解
2017/05/30 Javascript
NodeJS实现微信公众号关注后自动回复功能
2017/05/31 NodeJs
一个Js文件函数中调用另一个Js文件函数的方法演示
2017/08/14 Javascript
Vue 2.5.2下axios + express 本地请求404的解决方法
2018/02/21 Javascript
JS+CSS+HTML实现“代码雨”类似黑客帝国文字下落效果
2020/03/17 Javascript
微信小程序学习总结(五)常见问题实例小结
2020/06/04 Javascript
vue element实现表格合并行数据
2020/11/30 Vue.js
Python实现的检测web服务器健康状况的小程序
2014/09/17 Python
python对DICOM图像的读取方法详解
2017/07/17 Python
python链接oracle数据库以及数据库的增删改查实例
2018/01/30 Python
Linux下python3.6.1环境配置教程
2018/09/26 Python
运用Python的webbrowser实现定时打开特定网页
2019/02/21 Python
Django框架实现分页显示内容的方法详解
2019/05/10 Python
python的faker库用法
2019/11/28 Python
解决Python发送Http请求时,中文乱码的问题
2020/04/30 Python
大学毕业生自我鉴定
2013/11/05 职场文书
大学三年计划书范文
2014/04/30 职场文书
青奥会口号
2014/06/12 职场文书
机关干部四风问题自查报告及整改措施
2014/10/26 职场文书
python实现大文本文件分割成多个小文件
2021/04/20 Python
Windows 64位 安装 mysql 8.0.28 图文教程
2022/04/19 MySQL