Tensorflow自动删除旧模型文件

标签 tensorflow

我使用 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/

相关文章:

GPU 上的 Tensorflow 比 CPU 上的慢

python-3.x - 如何解决 "logits and labels must have the same first dimension"错误

高于 4 维张量的 Tensorflow keras BatchNormalization(视频输入)

python - 设置 LSTM 层的初始状态

python - 在 Keras 中 reshape 自定义损失函数中的张量

tensorflow - 如何在 Keras 中设置内核初始化程序 (glorot_uniform) 的种子值

tensorflow 小批量训练

2D 数组的 TensorFlow 卷积

python - 如何通过不规则索引获取子张量?

python - Tensorflow 中的预定采样