TensorFlow 检查点被覆盖

Question

我正在使用 Tensorflow 在输入集上训练模型（生成对抗网络），我想每 50 个 epoch 保存一次模型的参数。

假设我想训练模型 1000 个 epoch，并每 50 个 epoch 保存一次模型的参数，最终会有 20 个不同的检查点文件。

通过拥有一个 Session 和一个 Saver 对象，我只需使用以下代码即可。

if num_epoch % 50 == 0:
    saver.save(sess=sess, path='RGAN-1/sv/' + type_exp, global_step=num_epoch)

问题是，检查点被覆盖，在实验结束时，我只有最后 6 个检查点，而我应该有 20 个检查点。

我不知道为什么会这样。

Answer 1

tf.train.Saver 有一个 max_to_keep 参数，默认设置为 5。您可以通过 0 来保留所有检查点：

saver = tf.train.Saver(..., max_to_keep=0)

有关完整参数列表，请参阅 the docs。