解决TensorFlow训练模型及保存数量限制的问题


Posted in Python onMarch 03, 2021

每次卷积神经网络训练的结果都只保存了最后一部分,查阅了一下相关资料,发现是定义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 相关文章推荐
Python中的高级数据结构详解
Mar 27 Python
总结Python中逻辑运算符的使用
May 13 Python
python定时器(Timer)用法简单实例
Jun 04 Python
Python函数中不定长参数的写法
Feb 13 Python
Python基础知识点 初识Python.md
May 14 Python
python绘制BA无标度网络示例代码
Nov 21 Python
parser.add_argument中的action使用
Apr 20 Python
Python2与Python3关于字符串编码处理的差别总结
Sep 07 Python
python获得命令行输入的参数的两种方式
Nov 02 Python
selenium设置浏览器为headless无头模式(Chrome和Firefox)
Jan 08 Python
Django和Ueditor自定义存储上传文件的文件名
Feb 25 Python
python实现杨辉三角的几种方法代码实例
Mar 02 Python
使paramiko库执行命令时在给定的时间强制退出功能的实现
Mar 03 #Python
python上下文管理的使用场景实例讲解
Mar 03 #Python
pip install命令安装扩展库整理
Mar 02 #Python
python实现杨辉三角的几种方法代码实例
Mar 02 #Python
python爬虫破解字体加密案例详解
Mar 02 #Python
python包的导入方式总结
Mar 02 #Python
numpy实现RNN原理实现
Mar 02 #Python
You might like
JAVA/JSP学习系列之七
2006/10/09 PHP
做了CDN获取用户真实IP的函数代码(PHP与Asp设置方式)
2013/04/13 PHP
Php中使用Select 查询语句的实例
2014/02/19 PHP
基于laravel制作APP接口(API)
2016/03/15 PHP
PHP反射实际应用示例
2019/04/03 PHP
PHP如何使用array_unshift()在数组开头插入元素
2020/09/01 PHP
Add Formatted Data to a Spreadsheet
2007/06/12 Javascript
人人网javascript面试题 可以提前实现下
2012/01/05 Javascript
jquery与prototype框架的详细对比
2013/11/21 Javascript
搭建pomelo 开发环境
2014/06/24 Javascript
在JavaScript中使用JSON数据
2016/02/15 Javascript
使用getBoundingClientRect方法实现简洁的sticky组件的方法
2016/03/22 Javascript
JS 对象(Object)和字符串(String)互转方法
2016/05/20 Javascript
MUI 解决动态列表页图片懒加载再次加载不成功的bug问题
2017/04/13 Javascript
jQuery自动或手动图片切换效果
2017/10/11 jQuery
详解基于vue的服务端渲染框架NUXT
2018/06/20 Javascript
ES6中字符串的使用方法扩展
2019/06/04 Javascript
原生javascript单例模式的应用实例分析
2020/02/23 Javascript
[02:28]DOTA2亚洲邀请赛 LGD战队巡礼
2015/02/03 DOTA
Python的条件语句与运算符优先级详解
2015/10/13 Python
Python subprocess模块常见用法分析
2018/06/12 Python
Python发送邮件功能示例【使用QQ邮箱】
2018/12/04 Python
python求加权平均值的实例(附纯python写法)
2019/08/22 Python
tensorboard显示空白的解决
2020/02/15 Python
OpenCV灰度化之后图片为绿色的解决
2020/12/01 Python
阿里云:Aliyun.com
2017/02/15 全球购物
澳大利亚床上用品、浴巾和家居用品购物网站:Bambury
2020/04/16 全球购物
企业安全生产月活动总结
2014/07/05 职场文书
酒店销售经理岗位职责
2015/04/02 职场文书
倡议书的格式写法
2015/04/28 职场文书
主持稿开场白
2015/06/01 职场文书
生产车间管理制度
2015/08/04 职场文书
关于运动会的广播稿
2015/08/19 职场文书
2016年精神文明建设先进个人事迹材料
2016/02/29 职场文书
最新的离婚协议书范本!
2019/07/02 职场文书
Java 泛型详解(超详细的java泛型方法解析)
2021/07/02 Java/Android