解决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计数排序和基数排序算法实例
Apr 25 Python
python中ConfigParse模块的用法
Sep 29 Python
python读取json文件并将数据插入到mongodb的方法
Mar 23 Python
Python多线程编程简单介绍
Apr 13 Python
Python进阶篇之字典操作总结
Nov 16 Python
python机器学习库常用汇总
Nov 15 Python
Python如何处理大数据?3个技巧效率提升攻略(推荐)
Apr 15 Python
用Q-learning算法实现自动走迷宫机器人的方法示例
Jun 03 Python
Python pandas.DataFrame调整列顺序及修改index名的方法
Jun 21 Python
Python语法之精妙的十个知识点(装B语法)
Jan 18 Python
浅谈Keras的Sequential与PyTorch的Sequential的区别
Jun 17 Python
Python爬虫防封ip的一些技巧
Aug 06 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
在Windows系统上安装PHP运行环境文字教程
2010/07/19 PHP
PHP彩蛋信息介绍和阻止泄漏的方法(隐藏功能)
2014/08/06 PHP
PHP浮点比较大小的方法
2016/02/14 PHP
使用symfony命令创建项目的方法
2016/03/17 PHP
Yii2.0 Basic代码中路由链接被转义的处理方法
2016/09/21 PHP
浅谈PHP中静态方法和非静态方法的相互调用
2016/10/04 PHP
PHP实现深度优先搜索算法(DFS,Depth First Search)详解
2017/09/16 PHP
php/JS实现的生成随机密码(验证码)功能示例
2019/06/06 PHP
jquery连缀语法如何实现
2012/11/29 Javascript
jquery 层次选择器siblings与nextAll的区别介绍
2013/08/02 Javascript
JS操作数据库的实例代码
2013/10/17 Javascript
vue.js2.0点击获取自己的属性和jquery方法
2018/02/23 jQuery
jQuery中复合选择器简单用法示例
2018/03/31 jQuery
如何获取TypeScript的声明文件.d.ts
2018/05/01 Javascript
解决Idea、WebStorm下使用Vue cli脚手架项目无法使用Webpack别名的问题
2019/10/11 Javascript
Node.js API详解之 zlib模块用法分析
2020/05/19 Javascript
python通过加号运算符操作列表的方法
2015/07/28 Python
python浪漫表白源码
2019/04/05 Python
对Python 检查文件名是否规范的实例详解
2019/06/10 Python
python计算导数并绘图的实例
2020/02/29 Python
Django 拼接两个queryset 或是两个不可以相加的对象实例
2020/03/28 Python
使用Python实现批量ping操作方法
2020/05/06 Python
python3中编码获取网页的实例方法
2020/11/16 Python
python Pexpect模块的使用
2020/12/25 Python
HTML5实现的震撼3D焦点图动画的示例代码
2019/09/26 HTML / CSS
澳大利亚第一的设计师礼服租赁网站:GlamCorner
2017/08/13 全球购物
沙特阿拉伯电子产品和家用电器购物网站:Black Box
2019/07/24 全球购物
金鑫耀Java笔试题
2014/09/06 面试题
英文自荐信
2013/12/15 职场文书
优秀员工年终发言演讲稿
2014/01/01 职场文书
大学军训感言
2014/01/10 职场文书
淘宝中秋节活动方案
2014/01/31 职场文书
幼儿园父亲节活动方案
2014/03/11 职场文书
程序员求职信
2014/04/16 职场文书
国庆65周年演讲稿:回首往昔,展望未来
2014/09/21 职场文书
雷锋之歌观后感
2015/06/10 职场文书