解决TensorFlow训练模型及保存数量限制的问题
更新时间：2021年03月03日 08:36:35 作者：Destiny_Ren
这篇文章主要介绍了解决TensorFlow训练模型及保存数量限制的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
每次卷积神经网络训练的结果都只保存了最后一部分，查阅了一下相关资料，发现是定义saver时采用的默认值，这里进行如下设置：
saver = tf.train.Saver( max_to_keep = 100 , keep_checkpoint_every_n_hours = 1 )
补充：解决TensorFlow只能保存5个模型的问题
直奔主题
在训练模型的代码中找到这句代码：tf.train.Saver（），
改成：
tf.train.Saver(max_to_keep = m) # m为你想保存的模型数量
扩展
Saver类中的可选参数
tf.train.Saver(max_to_keep = m, keep_checkpoint_every_n_hours = n)
max_to_keep保存离当前训练最近的模型数量，默认值为5。如果想全部保存，并且电脑内存够用，设成多大都可以。
keep_checkpoint_every_n_hours每隔n个小时保存一次模型，默认值为10,000（一般情况下应该不会训练这么长时间，所以相当于是不会按照时间来保存，按照设置的epoch保存节点数来保存）。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。
相关文章
turtle库是一个点线面的简单图像库，在Python2.6之后被引入进来，能够完成一些比较简单的几何图像可视化。本文将利用turtle绘制一个可爱的草帽路飞，感兴趣的可以试一试2022-03-03
今天小编就为大家分享一篇Django 根据数据模型models创建数据表的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
这篇文章主要介绍了Python设置默认编码为utf8的方法,结合实例形式分析了Python针对文件编码的设置方法与相关注意事项,需要的朋友可以参考下2016-07-07
今天小编就为大家分享一篇python数据处理 根据颜色对图片进行分类的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
这篇文章主要介绍了利用PyQT5制作一个简单的二维码生成器，并打包成exe可执行程序。文中的示例代码讲解详细，感兴趣的同学可以了解一下2021-12-12
这篇文章主要介绍了使用Python的SymPy库解决数学运算问题的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-03-03
今天小编就为大家分享一篇使用Django搭建web服务器的例子(最最正确的方式)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
这篇文章主要介绍了Python文件操作基础流程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-03-03
这篇文章主要介绍了Python Sphinx使用实例及问题解决,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-01-01
Flask-RESTful是一个用于快速创建RESTful API接口的Flask扩展。，这篇文章主要介绍了flask-restful使用总结，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-12-12
最新评论