【发布时间】:2017-02-25 07:14:50
【问题描述】:
我使用 tf.train.Supervisor 开始一个会话来训练我的模型,并每 1000 步保存一次模型参数。但似乎 tensorflow 会自动删除旧模型文件。仅保存了 5 个最近的模型。生成 model.ckpt-6000 后,删除 model.ckpt-1000。但是我找不到关于这个操作的任何文件。
【问题讨论】:
标签: tensorflow
我使用 tf.train.Supervisor 开始一个会话来训练我的模型,并每 1000 步保存一次模型参数。但似乎 tensorflow 会自动删除旧模型文件。仅保存了 5 个最近的模型。生成 model.ckpt-6000 后,删除 model.ckpt-1000。但是我找不到关于这个操作的任何文件。
【问题讨论】:
标签: tensorflow
这里是文档: 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)
【讨论】:
tf.train.Saver(max_to_keep=None),以保存stackoverflow.com/a/38837404/5234494中指向的所有模型。