脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → TensorFlow训练模型保存数量限制

解决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保存节点数来保存）。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。

您可能感兴趣的文章:

相关文章

  • pyinstaller打包多个py文件和去除cmd黑框的方法

    pyinstaller打包多个py文件和去除cmd黑框的方法

    今天小编就为大家分享一篇pyinstaller打包多个py文件和去除cmd黑框的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06
  • python正则表达式之re.match()与re.search()的用法及区别

    python正则表达式之re.match()与re.search()的用法及区别

    本文主要介绍了python正则表达式之re.match()与re.search()的用法及区别，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2022-08-08
  • 深入探究Python Numba库编译优化利器

    深入探究Python Numba库编译优化利器

    这篇文章主要为大家介绍了Python Numba库编译优化利器深入探究,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2024-01-01
  • 如何基于pandas读取csv后合并两个股票

    如何基于pandas读取csv后合并两个股票

    这篇文章主要介绍了如何基于pandas读取csv后合并两个股票,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-09-09
  • python继承threading.Thread实现有返回值的子类实例

    python继承threading.Thread实现有返回值的子类实例

    这篇文章主要介绍了python继承threading.Thread实现有返回值的子类实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-05-05
  • Qt通过QGraphicsview实现简单缩放及还原效果

    Qt通过QGraphicsview实现简单缩放及还原效果

    本文主要介绍通过QGraphicsview实现简单的缩放以及缩放后还原原始大小，通过scale可以对view进行放大或缩小，具体内容详情跟随小编一起看看吧
    2021-09-09
  • 对python中的*args与**kwgs的含义与作用详解

    对python中的*args与**kwgs的含义与作用详解

    今天小编就为大家分享一篇对python中的*args与**kwgs的含义与作用详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-08-08
  • Python爬虫爬取网站图片

    Python爬虫爬取网站图片

    这篇文章主要介绍了Python爬虫爬取网站图片,此次python3主要用requests，解析图片网址主要用beautiful soup，可以基本完成爬取图片功能,需要的朋友可以参考一下
    2022-01-01
  • Centos环境部署django项目的全过程(永久复用)

    Centos环境部署django项目的全过程(永久复用)

    Django是一款针对Python环境的WEB开发框架,能够帮助我们构架快捷,下面这篇文章主要给大家介绍了关于Centos环境部署django项目的相关资料,需要的朋友可以参考下
    2022-10-10
  • Python CSV模块使用实例

    Python CSV模块使用实例

    这篇文章主要介绍了Python CSV模块使用实例,本文将举几个例子来介绍一下Python的CSV模块的使用方法,包括reader、writer、DictReader、DictWriter.register_dialect等,需要的朋友可以参考下
    2015-04-04

最新评论