解决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的Django框架中包装视图函数
Jul 20 Python
使用Python3制作TCP端口扫描器
Apr 17 Python
Python常用时间操作总结【取得当前时间、时间函数、应用等】
May 11 Python
python万年历实现代码 含运行结果
May 20 Python
python 对给定可迭代集合统计出现频率,并排序的方法
Oct 18 Python
在Python中定义一个常量的方法
Nov 10 Python
DES加密解密算法之python实现版(图文并茂)
Dec 06 Python
Python制作动态字符图的实例
Jan 27 Python
Python实现 PS 图像调整中的亮度调整
Jun 28 Python
Numpy与Pytorch 矩阵操作方式
Dec 27 Python
matplotlib subplot绘制多个子图的方法示例
Jul 28 Python
Matplotlib绘制混淆矩阵的实现
May 27 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
php4的session功能评述(一)
2006/10/09 PHP
DW中链接mysql数据库时,建立字符集中文出现乱码的解决方法
2010/03/27 PHP
php 团购折扣计算公式
2011/11/24 PHP
深入PHP nl2br()格式化输出的详解
2013/06/05 PHP
项目中应用Redis+Php的场景
2016/05/22 PHP
使用jQuery UI的tooltip函数修饰title属性的气泡悬浮框
2013/06/24 Javascript
asp.net刷新本页面的六种方法总结
2014/01/07 Javascript
浏览器窗口加载和大小改变事件示例
2014/02/27 Javascript
jquery实现的一个简单进度条效果实例
2014/05/12 Javascript
javascript返回顶部的按钮实现方法
2016/01/09 Javascript
js实现html table 行,列锁定的简单实例
2016/10/13 Javascript
常用的js方法合集
2017/03/10 Javascript
Centos6.8下Node.js安装教程
2017/05/12 Javascript
微信小程序 转发功能的实现
2017/08/04 Javascript
nodejs acl的用户权限管理详解
2018/03/14 NodeJs
jquery中为什么能用$操作
2019/06/18 jQuery
解决Echarts2竖直datazoom滑动后显示数据不全的问题
2020/07/20 Javascript
[00:10]DOTA2全国高校联赛速递
2018/05/30 DOTA
[50:21]Liquid vs Winstrike 2018国际邀请赛小组赛BO2 第二场
2018/08/19 DOTA
python中MySQLdb模块用法实例
2014/11/10 Python
TensorFlow实现创建分类器
2018/02/06 Python
python生成1行四列全2矩阵的方法
2018/08/04 Python
详解Python 调用C# dll库最简方法
2019/06/20 Python
用Pelican搭建一个极简静态博客系统过程解析
2019/08/22 Python
使用matplotlib绘制图例标签中带有公式的图
2019/12/13 Python
html5中去掉input type date默认样式的方法
2018/09/06 HTML / CSS
英国男女奢华内衣和泳装购物网站:Figleaves
2017/01/28 全球购物
澳大利亚珠宝商:Shiels
2019/10/06 全球购物
大学生入党思想汇报
2014/01/14 职场文书
《我的第一本书》教学反思
2014/02/15 职场文书
大学生自我鉴定书
2014/03/24 职场文书
党员创先争优活动总结
2014/05/04 职场文书
社区法制宣传月活动总结
2015/05/07 职场文书
幼儿园中班班级总结
2015/08/10 职场文书
Qt自定义Plot实现曲线绘制的详细过程
2021/11/02 Python
python中pycryto实现数据加密
2022/04/29 Python