基于tensorflow加载部分层的方法
更新时间：2018年07月26日 09:57:55 作者：jinglingli_SJTU
今天小编就为大家分享一篇基于tensorflow加载部分层的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
一般使用
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加载部分层的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
在自然语言处理界，模式匹配可以说是最常用的技术。甚至可以说，将NLP技术作为真实生产力的项目都少不了模式匹配。本文就介绍了pampy超强的模式匹配工具的实现，感兴趣的可以了解一下2021-07-07
这篇文章主要为大家详细介绍了Python如何使用Matplotlib绘制3D圣诞树,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下2023-12-12
这篇文章主要介绍了Python编程实战之Oracle数据库操作,结合具体实例形式分析了Python的Oracle数据库模块cx_Oracle包安装、Oracle连接及操作技巧,需要的朋友可以参考下2017-06-06
今天小编就为大家分享一篇python logging设置level失败的解决方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
这篇文章主要为大家详细介绍了python书籍信息爬虫示例，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-03-03
Python实现对Excel文件中不在指定区间内的数据加以去除的方法
这篇文章主要介绍了基于Python语言,读取Excel表格文件,基于我们给定的规则,对其中的数据加以筛选,将不在指定数据范围内的数据剔除,保留符合我们需要的数据的方法,需要的朋友可以参考下2023-08-08
这篇文章主要介绍了keras模型保存为tensorflow的二进制模型方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
这篇文章主要介绍了爬山算法,爬山法（climbing method）是一种优化算法，其一般从一个随机的解开始，然后逐步找到一个最优解（局部最优）然后用Python实现了这个算法,需要的朋友可以参考下2014-04-04
这篇文章主要为大家详细介绍了Python使用Pillow添加水印，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-05-05
这篇文章主要介绍了详解Python locals()的陷阱，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-03-03
最新评论