基于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加载部分层的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • 解决c++调用python中文乱码问题

    解决c++调用python中文乱码问题

    这篇文章主要介绍了c++调用python中文乱码问题,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-07-07
  • pyinstaller打包单个exe后无法执行错误的解决方法

    pyinstaller打包单个exe后无法执行错误的解决方法

    今天小编就为大家分享一篇pyinstaller打包单个exe后无法执行错误的解决方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06
  • 带你了解Python妙开根号的三种方式

    带你了解Python妙开根号的三种方式

    这篇文章主要为大家介绍了Python妙开根号的三种方式,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助
    2022-01-01
  • python通过scapy获取局域网所有主机mac地址示例

    python通过scapy获取局域网所有主机mac地址示例

    这篇文章主要介绍了python通过scapy获取局域网所有主机mac地址示例,需要的朋友可以参考下
    2014-05-05
  • Python实现将字典内容保存到Excel文件的四种方法

    Python实现将字典内容保存到Excel文件的四种方法

    Python 中的字典是一个数据集合,其中每个值对应一个键,它们是无序的、可变的,并且对字典中存储的值和键的数据类型没有限制,本文给大家介绍了Python实现将字典内容保存到Excel文件的四种方法,需要的朋友可以参考下
    2024-07-07
  • 如何使用Python生成4位数的随机数字

    如何使用Python生成4位数的随机数字

    本文讨论了如何使用randint() 和randrange() 方法来生成一个四位数的数字,此外,我们还讨论了另一种拥有随机四位数号码的途径,感兴趣的朋友跟随小编一起看看吧
    2023-10-10
  • 使用Pandas的ExcelWriter操作excel的方法

    使用Pandas的ExcelWriter操作excel的方法

    这篇文章主要介绍了使用Pandas的ExcelWriter操作excel的方法,ExcelWriter这个插件有个坑,就是已经设置好的格式是无法更改的,因此,由pandas转成excel的时候,必须将格式清除,尤其是表头的格式需要大家多多注意,本文结合示例代码讲解的非常详细,需要的朋友参考下吧
    2023-11-11
  • Pyecharts中的饼图位置调整方式

    Pyecharts中的饼图位置调整方式

    这篇文章主要介绍了Pyecharts 饼图位置调整方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2022-11-11
  • Python中文文本处理利器jieba分词库使用

    Python中文文本处理利器jieba分词库使用

    这篇文章主要给大家介绍了关于Python中文文本处理利器jieba分词库使用的相关资料,jieba是python中一个重要的第三方中文分词函数库,文中通过代码介绍的非常详细,需要的朋友可以参考下
    2023-12-12
  • python可视化分析绘制散点图和边界气泡图

    python可视化分析绘制散点图和边界气泡图

    这篇文章主要介绍了python可视化分析绘制散点图和边界气泡图,python绘制散点图,展现两个变量间的关系,当数据包含多组时,使用不同颜色和形状区分
    2022-06-06

最新评论