基于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 相关文章推荐
python3使用urllib示例取googletranslate(谷歌翻译)
Jan 23 Python
Python正规则表达式学习指南
Aug 02 Python
Python 中 Virtualenv 和 pip 的简单用法详解
Aug 18 Python
python异步实现定时任务和周期任务的方法
Jun 29 Python
Tornado实现多进程/多线程的HTTP服务详解
Jul 25 Python
Python3常见函数range()用法详解
Dec 30 Python
tensorflow 模型权重导出实例
Jan 24 Python
python调用私有属性的方法总结
Jul 24 Python
Python容器类型公共方法总结
Aug 19 Python
Python WebSocket长连接心跳与短连接的示例
Nov 24 Python
windows安装python超详细图文教程
May 21 Python
用Python实现屏幕截图详解
Jan 22 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
PHP生成各种随机验证码的方法总结【附demo源码】
2017/06/05 PHP
php图片合成方法(多张图片合成一张)
2017/11/25 PHP
one.php 多项目、函数库、类库 统一为一个版本的方法
2020/08/24 PHP
细品javascript 寻址,闭包,对象模型和相关问题
2009/04/27 Javascript
基于jQuery的淡入淡出可自动切换的幻灯插件
2010/08/24 Javascript
javascript右下角弹层及自动隐藏(自己编写)
2013/11/20 Javascript
深入理解JavaScript系列(46):代码复用模式(推荐篇)详解
2015/03/04 Javascript
谈谈Jquery ajax中success和complete有哪些不同点
2015/11/20 Javascript
VUE axios上传图片到七牛的实例代码
2017/07/28 Javascript
关于HTTP传输中gzip压缩的秘密探索分析
2018/01/12 Javascript
jQuery实现checkbox全选功能完整实例
2018/07/12 jQuery
vue 监听屏幕高度的实例
2018/09/05 Javascript
JavaScript闭包与作用域链实例分析
2019/01/21 Javascript
node.js实现微信开发之获取用户授权
2019/03/18 Javascript
微信小程序云开发如何实现数据库自动备份实现
2019/08/16 Javascript
ionic+html5+API实现双击返回键退出应用
2019/09/17 Javascript
javascript实现视频弹幕效果(两个版本)
2019/11/28 Javascript
[44:39]2014 DOTA2国际邀请赛中国区预选赛 NE VS CNB
2014/05/21 DOTA
[02:59]DOTA2完美大师赛主赛事第三日精彩集锦
2017/11/25 DOTA
Python中endswith()函数的基本使用
2015/04/07 Python
Django权限机制实现代码详解
2018/02/05 Python
Ubuntu16.04/树莓派Python3+opencv配置教程(分享)
2018/04/02 Python
解决Ubuntu pip 安装 mysql-python包出错的问题
2018/06/11 Python
flask框架路由常用定义方式总结
2019/07/23 Python
Pytorch 实现计算分类器准确率(总分类及子分类)
2020/01/18 Python
Django windows使用Apache实现部署流程解析
2020/10/12 Python
HTML5之语义标签介绍
2016/07/07 HTML / CSS
Internet主要有哪些网络群组成
2015/12/24 面试题
如果重写了对象的equals()方法,需要考虑什么
2014/11/02 面试题
Java文件和目录(IO)操作
2014/08/26 面试题
小学教师师德整改措施
2014/09/29 职场文书
2016大一新生军训心得体会
2016/01/11 职场文书
2019最新版劳务派遣管理制度
2019/08/16 职场文书
nginx常用配置conf的示例代码详解
2022/03/21 Servers
PostgreSQL 插入INSERT、删除DELETE、更新UPDATE、事务transaction
2022/04/12 PostgreSQL
PostgreSQL数据库去除重复数据和运算符的基本查询操作
2022/04/12 PostgreSQL