【问题标题】:Tensorflow automatically deletes old model filesTensorFlow 自动删除旧模型文件
【发布时间】:2017-02-25 07:14:50
【问题描述】:

我使用 tf.train.Supervisor 开始一个会话来训练我的模型,并每 1000 步保存一次模型参数。但似乎 tensorflow 会自动删除旧模型文件。仅保存了 5 个最近的模型。生成 model.ckpt-6000 后,删除 model.ckpt-1000。但是我找不到关于这个操作的任何文件。

【问题讨论】:

    标签: tensorflow


    【解决方案1】:

    这里是文档: https://www.tensorflow.org/api_docs/python/tf/train/Saver

    它有一个名为 max_to_keep 的变量,默认为 5。所以只需调用

    tf.train.Saver(max_to_keep=100000) 
    

    或其他包含所有可能创建的大数字。无法将所有模型保存为我所知道的选项。

    更新:正如 Ferran Parés 在评论中指出的那样,只需设置 max_to_keep=None 即可保存所有输出。

    tf.train.Saver(max_to_keep=None) 
    

    【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2016-10-03
    • 2019-03-03
    • 1970-01-01
    • 2018-09-27
    • 2012-03-07
    • 1970-01-01
    相关资源
    最近更新 更多