如何解决tensorflow恢复模型的特定值时出错

 更新时间:2020年02月06日 14:50:13   作者:深度学习小学生  
今天小编就为大家分享一篇如何解决tensorflow恢复模型的特定值时出错,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

模型的恢复

对于的模型的恢复来说,需要首先恢复模型的整个图文件,之后从图文件中读取相应的节点信息。

存储的模型文件包括四个子文件,如下: 

现在假如我想恢复模型中的某个节点信息:

需要注意的是在使用saver.restore恢复权值的时候,参数的设置需要万分注意:

# 先加载图文件
saver = tf.train.import_meta_graph("./model/save_model.meta")
graph = tf.get_default_graph()
# 其中的一个节点
a_val = graph.get_tensor_by_name("var/a_val:0") # 这里是我的节点名称,这都不是重点

with tf.Session() as sess:
  # 加载权值
  saver.restore(sess, "./model/save_model") # 这里才是重点,注意使用的是save_model,而不是save_model.data-00000-of-00001,不要带一串后缀,如果使用后者的话,会报错!!!!!
  print(sess.run(a_val))

以上这篇如何解决tensorflow恢复模型的特定值时出错就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • 使用Django和Flask获取访问来源referrer

    使用Django和Flask获取访问来源referrer

    这篇文章主要介绍了使用Django和Flask获取访问来源referrer,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-04-04
  • Python实现大文件排序的方法

    Python实现大文件排序的方法

    这篇文章主要介绍了Python大文件排序的方法,涉及Python针对文件、缓存及日期等操作的相关技巧,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-07-07
  • python深度学习之多标签分类器及pytorch实现源码

    python深度学习之多标签分类器及pytorch实现源码

    这篇文章主要为大家介绍了python深度学习之多标签分类器的使用说明及pytorch的实现源码,有需要的朋友可以借鉴参考下,希望能够有所帮助
    2022-01-01
  • python在控制台输出进度条的方法

    python在控制台输出进度条的方法

    这篇文章主要介绍了python在控制台输出进度条的方法,实例分析了Python输出进度条效果的方法,需要的朋友可以参考下
    2015-06-06
  • Python 导入文件过程图解

    Python 导入文件过程图解

    这篇文章主要介绍了Python 导入文件过程图解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-10-10
  • Python数据分析 Pandas Series对象操作

    Python数据分析 Pandas Series对象操作

    这篇文章主要介绍了Python数据分析之Pandas Series对象,文章基于python的相关资料展开详细内容介绍,具有一定的参考价值,需要的小伙伴可以参考一下
    2022-05-05
  • 使用Python的SymPy库解决数学运算问题的方法

    使用Python的SymPy库解决数学运算问题的方法

    这篇文章主要介绍了使用Python的SymPy库解决数学运算问题的方法,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2019-03-03
  • python脚本定时发送邮件

    python脚本定时发送邮件

    这篇文章主要为大家详细介绍了python脚本定时发送邮件,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2020-12-12
  • Python高效编程技巧

    Python高效编程技巧

    我已经使用Python编程有多年了,即使今天我仍然惊奇于这种语言所能让代码表现出的整洁和对DRY编程原则的适用。这些年来的经历让我学到了很多的小技巧和知识,大多数是通过阅读很流行的开源软件,如Django, Flask, Requests中获得的
    2013-01-01
  • python关于调用函数外的变量实例

    python关于调用函数外的变量实例

    今天小编就为大家分享一篇python关于调用函数外的变量实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-12-12

最新评论