解决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保存节点数来保存）。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。
相关文章
在本篇文章里小编给大家分享的是一篇关于Python编写万花尺图案实例的内容，有兴趣的朋友们可以参考下。2021-01-01
这篇文章主要为大家详细介绍了python实现图像增强算法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-08-08
今天小编就为大家分享一篇Pytorch evaluation每次运行结果不同的解决，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
这篇文章主要介绍了Python3.5面向对象编程,结合图文与实例形式详细分析了Python面向对象编程相关的概念、类定义、实例化、实例变量、类变量、析构函数等相关原理及使用技巧,需要的朋友可以参考下2019-04-04
这篇文章主要介绍了Python 如何优雅的将数字转化为时间格式的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-09-09
这篇文章主要介绍了Python OS模块,结合实例形式总结分析了Python使用OS解析文件路径、判断文件、目录等相关操作技巧,需要的朋友可以参考下2019-04-04
箭头向量图十分常见，比如天气预报在显示风场的时候，就会贴心地用箭头指明风的方向，下面就跟随小编一起学习一下如何利用Python绘制箭头向量图吧2023-08-08
Python最强大的功能之一是能够十分方便地使用性能优异的第三方库,这些库涵盖了大量的不同领域,下面这篇文章主要给大家介绍了关于利用pip安装python第三方库的4种方法,需要的朋友可以参考下2023-02-02
这篇文章主要为大家介绍了python的循环，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2021-12-12
Django框架会话技术实例分析【Cookie与Session】
这篇文章主要介绍了Django框架会话技术,结合实例形式分析了Django框架Cookie与Session相关使用技巧与注意事项,需要的朋友可以参考下2019-05-05
最新评论