TensorFlow train.Supervisor - 在訓練停止時保存檢查點？

Question

在TensorFlow 1.0中，tf.train.Supervisor以save_model_secs的間隔保存檢查點。在培訓結束時是否有辦法保存檢查點，而不是定期在培訓期間？

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))