浅谈tf.train.Saver()与tf.train.import_meta_graph的要点
(一)、tf.train.Saver()
(1). tf.train.Saver() 是用来保存tensorflow训练模型的，默认保存全部参数
(2). 用来加载参数，注：只加载存储在data中的权重和偏置项等需要训练的参数，其他一律不加载，包括meta文件中的图也不加载
(二)、tf.train.import_meta_graph
(1). 用来加载meta文件中的图,以及图上定义的结点参数包括权重偏置项等需要训练的参数,也包括训练过程生成的中间参数,所有参数都是通过graph调用接口get_tensor_by_name(name="训练时的参数名称")来获取
(三)、总结
(1). 保存使用tf.train.Saver()
(2). 加载可以使用tf.train.import_meta_graph(".meta文件"),直接通过训练参数名称就可以获取需要的参数,但需要提前知道训练时的参数名称才能获取,要懂得tensorflow命名规则
(3). tf.train.Saver("./checkpoints目录/")加载的缺点是只加载了训练参数,并且必须定义与之相同(shape,dtype要相同,tf.type要相同，如：我是placeholder,你也是tf.placeholder)方能使用,当你要获取训练中间参数时,需要和训练过程一样搭建相同的网络.
补充：tf.train.import_meta_graph报KeyError
我在模型恢复时，在执行tf.train.import_meta_graph的时候报错
后来发现，我的模型是在服务器上训练的，服务器上tensorflow版本是1.11.0,而我在本地电脑上执行的 tf.train.import_meta_graph，我本地的tensorflow是1.5.0，我将tensorflow更新到1.11.0后，就解决了。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python读写txt文本文件的操作方法全解析,包括对文本的查找和替换等技巧的讲解,需要的朋友可以参考下2016-06-06
这篇文章主要为大家详细介绍了python实现定时压缩指定文件夹发送邮件，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-04-04
本文主要介绍了python httpx的具体使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
Pycharm运行时总是跳出Python Console问题
这篇文章主要介绍了Pycharm运行时总是跳出Python Console问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-04-04
今天小编就为大家分享一篇在PyCharm中批量查找及替换的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
这篇文章主要为大家介绍了解决90%的常见问题的8个python NumPy函数示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-06-06
这篇文章主要介绍了Python+OpenCV 实现高斯滤波的过程,关于高斯滤波的解释，它是一种线性平滑滤波，适用于消除高斯噪声，具体实现过程跟随小编一起看看吧2021-10-10
这篇文章主要介绍了Python自动化测试selenium指定截图文件名方法，Selenium 支持 Web 浏览器的自动化，它提供一套测试函数，用于支持 Web 自动化测试，下文基于python实现指定截图文件名方法，需要的小伙伴可以参考一下2022-05-05
数据可视化是任何探索性数据分析或报告的关键步骤，目前有许多非常好的商业智能工具，比如Tableau、googledatastudio和PowerBI等，本文就详细的进行对比，感兴趣的可以了解一下2021-06-06
在本篇文章里小编给大家整理的是一篇关于postman传递当前时间戳知识点相关内容，有需要的朋友们可以学习下。2019-09-09
最新评论