我使用 tf.train.Supervisor 启动一个 session 来训练我的模型,并每 1000 步保存一次模型参数。但似乎tensorflow会自动删除旧的模型文件。仅保存了 5 个最近的模型。一旦 model.ckpt-6000 生成,model.ckpt-1000 就会被删除。但我找不到任何有关此操作的文档。
最佳答案
以下是文档: 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)
关于Tensorflow自动删除旧模型文件,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42453138/