基于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 with用法实例
Apr 14 Python
使用Python中的tkinter模块作图的方法
Feb 07 Python
python使用knn实现特征向量分类
Dec 26 Python
python3爬虫学习之数据存储txt的案例详解
Apr 24 Python
pytorch 固定部分参数训练的方法
Aug 17 Python
pandas中ix的使用详细讲解
Mar 09 Python
详解Ubuntu环境下部署Django+uwsgi+nginx总结
Apr 02 Python
python如何求100以内的素数
May 27 Python
Python中bisect的用法及示例详解
Jul 20 Python
解决python3安装pandas出错的问题
May 20 Python
Django实现翻页的示例代码
May 24 Python
python中pymysql包操作数据库方法
Apr 19 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在Web开发领域的优势
2006/10/09 PHP
PHP设计模式 注册表模式(多个类的注册)
2012/02/05 PHP
百度地图API使用方法详解
2015/08/25 PHP
分享PHP计算两个日期相差天数的代码
2015/12/23 PHP
PHP基于phpqrcode类生成二维码的方法示例详解
2020/08/07 PHP
onkeydown事件解决按回车键直接提交数据的需求
2013/04/11 Javascript
jQuery随便控制任意div隐藏的方法
2013/06/28 Javascript
JS调用CS里的带参方法实例
2013/08/01 Javascript
js获取指定日期前后的日期代码
2013/08/20 Javascript
原生js操作checkbox用document.getElementById实现
2013/10/12 Javascript
jquery实现显示已选用户
2014/07/21 Javascript
JavaScript函数作用域链分析
2015/02/13 Javascript
jQuery入门之层次选择器实例简析
2015/12/11 Javascript
angularjs ui-router中路由的二级嵌套
2017/03/10 Javascript
jQuery+CSS实现的table表格行列转置功能示例
2018/01/08 jQuery
jQuery中复合选择器简单用法示例
2018/03/31 jQuery
浅谈vue项目4rs vue-router上线后history模式遇到的坑
2018/09/27 Javascript
详解如何运行vue项目
2019/04/15 Javascript
小程序实现短信登录倒计时
2019/07/12 Javascript
VUE中使用HTTP库Axios方法详解
2020/02/05 Javascript
vue中echarts引入中国地图的案例
2020/07/28 Javascript
详解vue路由
2020/08/05 Javascript
Python优化技巧之利用ctypes提高执行速度
2016/09/11 Python
Python中的函数作用域
2018/05/07 Python
Python 下载及安装详细步骤
2019/11/04 Python
PyCharm 无法 import pandas 程序卡住的解决方式
2020/03/09 Python
Python如何设置指定窗口为前台活动窗口
2020/08/12 Python
Currentbody法国:健康与美容高科技产品
2020/08/16 全球购物
请解释流与文件有什么不同
2016/07/29 面试题
js实现弹框效果
2021/03/24 Javascript
松材线虫病防治方案
2014/06/15 职场文书
中专生自荐信
2014/06/25 职场文书
社区领导班子四风问题原因分析及整改措施
2014/09/28 职场文书
离婚协议书怎样才有法律效力
2014/10/10 职场文书
单方离婚协议书范本2014
2014/10/28 职场文书
绵山导游词
2015/02/05 职场文书