解决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实现比较两个列表(list)范围
Jun 12 Python
Python的Flask框架应用程序实现使用QQ账号登录的方法
Jun 07 Python
Python入门之三角函数atan2()函数详解
Nov 08 Python
python编程线性回归代码示例
Dec 07 Python
在python 中split()使用多符号分割的例子
Jul 15 Python
简单了解python变量的作用域
Jul 30 Python
Python爬虫 scrapy框架爬取某招聘网存入mongodb解析
Jul 31 Python
python excel转换csv代码实例
Aug 26 Python
Python 如何提高元组的可读性
Aug 26 Python
python pillow模块使用方法详解
Aug 30 Python
用python写测试数据文件过程解析
Sep 25 Python
Pandas读取csv时如何设置列名
Jun 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
在wamp集成环境下升级php版本(实现方法)
2013/07/01 PHP
thinkphp3.x中display方法及show方法的用法实例
2016/05/19 PHP
PHP实现蛇形矩阵,回环矩阵及数字螺旋矩阵的方法分析
2017/05/29 PHP
php计数排序算法的实现代码(附四个实例代码)
2020/03/31 PHP
js 页面传参数时 参数值含特殊字符的问题
2009/12/13 Javascript
JavaScript 学习笔记(七)字符串的连接
2009/12/31 Javascript
JS事件在IE与FF中的区别详细解析
2013/11/20 Javascript
js实现表格字段排序
2014/02/19 Javascript
JQuery1.8 判断元素是否绑定事件的方法
2014/07/10 Javascript
javascript结合Canvas 实现简易的圆形时钟
2015/03/11 Javascript
JavaScript实现信用卡校验方法
2015/04/07 Javascript
javascript日期计算实例分析
2015/06/29 Javascript
JQuery中ajax方法访问web服务实例
2015/07/18 Javascript
JavaScript知识点总结(十六)之Javascript闭包(Closure)代码详解
2016/05/31 Javascript
AngularJs Javascript MVC 框架
2016/06/20 Javascript
相册展示PhotoSwipe.js插件实现
2016/08/25 Javascript
AngularJs  E2E Testing 详解
2016/09/02 Javascript
JS给Array添加是否包含字符串的简单方法
2016/10/29 Javascript
vue2滚动条加载更多数据实现代码
2017/01/10 Javascript
express框架实现基于Websocket建立的简易聊天室
2017/08/10 Javascript
微信小程序实现单选功能
2018/10/30 Javascript
javascript前端实现多视频上传
2020/12/13 Javascript
解决Python3中的中文字符编码的问题
2018/07/18 Python
numpy 矩阵形状调整:拉伸、变成一位数组的实例
2020/06/18 Python
CSS3 Columns分列式布局方法简介
2014/05/03 HTML / CSS
JD Sports澳洲官网:英国领先的运动鞋和运动时尚零售商
2020/02/15 全球购物
Trench London官方网站:高级风衣和意大利皮夹克
2020/07/11 全球购物
运动会的口号
2014/06/09 职场文书
英语自我介绍演讲稿
2014/09/01 职场文书
食品安全演讲稿
2014/09/01 职场文书
伊索寓言读书笔记
2015/06/30 职场文书
听课评课活动心得体会
2016/01/15 职场文书
班组长如何制订适合本班组的工作计划?
2019/07/10 职场文书
JS监听Esc 键触发事键
2021/04/14 Javascript
如何使用php生成zip压缩包
2021/04/21 PHP
python图像处理 PIL Image操作实例
2022/04/09 Python