TensorFlow train.Supervisor - 在训练停止时保存检查点？

Question

在TensorFlow 1.0中，save_model_secs以letters <- "abc" numbers <- "123" mix <- "b1dd" 为间隔保存检查点。有没有办法在训练结束时保存检查点，而不是在训练期间定期保存？

Answer 1

tf.train.Supervisor在循环结束时写入检查点。如果您想避免编写其他检查点，可以将save_model_secs设置为较大的值。这是一个只保存单个最终检查点的示例：

import tensorflow as tf

y = tf.Variable(0)
y = tf.assign_add(y, 1)

sv = tf.train.Supervisor(logdir='/tmp', save_model_secs=100000000)

with sv.managed_session() as sess:

    for step in range(10):
        if sv.should_stop():
             break

        print(sess.run(y))