基于tensorflow加载部分层的方法


Posted in Python onJuly 26, 2018

一般使用

saver.restore(sess, modeldir + "model.ckpt")

即可加载已经训练好的网络,可是有时候想值使用部分层的参数,这时候可以选择在加载网络之后重新初始化剩下的层

var_list = [weights['wd1'], weights['out'], biases['bd1'], biases['out'], global_step]
initfc = tf.variables_initializer(var_list, name='init')

比如我们想从新初始化var_list中的各个层,在restore之后,再初始化即可

sess.run(init)
saver.restore(sess, modeldir + "model.ckpt")
print sess.run(global_step)
#initialize several layer
sess.run(initfc)
print sess.run(global_step)

即可发现部分变量重新初始化了

以上这篇基于tensorflow加载部分层的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python基于列表list实现的CRUD操作功能示例
Jan 05 Python
Python tornado队列示例-一个并发web爬虫代码分享
Jan 09 Python
TensorFlow高效读取数据的方法示例
Feb 06 Python
python方法生成txt标签文件的实例代码
May 10 Python
Python实现的读取文件内容并写入其他文件操作示例
Apr 09 Python
python中break、continue 、exit() 、pass终止循环的区别详解
Jul 08 Python
Python代理IP爬虫的新手使用教程
Sep 05 Python
python的time模块和datetime模块实例解析
Nov 29 Python
pytorch构建多模型实例
Jan 15 Python
Pytorch数据拼接与拆分操作实现图解
Apr 30 Python
pytorch 预训练模型读取修改相关参数的填坑问题
Jun 05 Python
python元组打包和解包过程详解
Aug 02 Python
利用python画出折线图
Jul 26 #Python
浅谈flask源码之请求过程
Jul 26 #Python
python画折线图的程序
Jul 26 #Python
TensorFlow利用saver保存和提取参数的实例
Jul 26 #Python
78行Python代码实现现微信撤回消息功能
Jul 26 #Python
python opencv实现图片旋转矩形分割
Jul 26 #Python
Flask之flask-session的具体使用
Jul 26 #Python
You might like
哪吒敖丙传:新人物二哥敖乙出场 小敖丙奶气十足
2020/03/08 国漫
PHP生成各种随机验证码的方法总结【附demo源码】
2017/06/05 PHP
PHP-FPM和Nginx的通信机制详解
2019/02/01 PHP
PHP模版引擎原理、定义与用法实例
2019/03/29 PHP
laravel框架创建授权策略实例分析
2019/11/22 PHP
Jquery chosen动态设置值实例介绍
2013/08/08 Javascript
setInterval与clearInterval的使用示例代码
2014/01/28 Javascript
jQuery动态背景图片效果实现方法
2015/07/03 Javascript
jquery图片倾斜层叠切换特效代码分享
2015/08/27 Javascript
nodejs初步体验篇
2015/11/23 NodeJs
各式各样的导航条效果css3结合jquery代码实现
2016/09/17 Javascript
有关文件上传 非ajax提交 得到后台数据问题
2016/10/12 Javascript
vue多级多选菜单组件开发
2020/09/08 Javascript
原生js实现可爱糖果数字时间特效
2016/12/30 Javascript
用JS实现简单的登录验证功能
2017/07/28 Javascript
基于Koa2写个脚手架模拟接口服务的方法
2018/11/27 Javascript
浅析vue中的provide / inject 有什么用处
2019/11/10 Javascript
[16:27]DOTA2 HEROS教学视频教你分分钟做大人-艾欧
2014/06/11 DOTA
python中sets模块的用法实例
2014/09/30 Python
详解Python的Flask框架中生成SECRET_KEY密钥的方法
2016/06/07 Python
python基于itchat实现微信群消息同步机器人
2017/02/27 Python
Python3利用SMTP协议发送E-mail电子邮件的方法
2017/09/30 Python
将Django项目部署到CentOs服务器中
2018/10/18 Python
Python基于plotly模块实现的画图操作示例
2019/01/23 Python
python读取有密码的zip压缩文件实例
2019/02/08 Python
Python aiohttp百万并发极限测试实例分析
2019/10/26 Python
浅谈python中频繁的print到底能浪费多长时间
2020/02/21 Python
Hotels.com台湾:饭店订房网
2017/09/06 全球购物
俄罗斯鲜花递送:AMF
2020/04/24 全球购物
高二化学教学反思
2014/01/30 职场文书
社区护士演讲稿
2014/08/27 职场文书
第二批党的群众路线教育实践活动个人对照检查材料
2014/09/23 职场文书
银行柜员与客户起冲突检讨书
2014/09/27 职场文书
南阳市白酒市场的调查报告
2019/11/08 职场文书
vite+vue3.0+ts+element-plus快速搭建项目的实现
2021/06/24 Vue.js
Win11安装升级时提示“该电脑必须支持安全启动”
2022/04/19 数码科技