受监控的培训课程保存所有检查点

Him*_*ghi 2 tensorflow

使用时tf.train.MonitoredTrainingSession,可以保存所有检查点。它有一个参数(save_checkpoint_secs=600)来指定要保存多少个检查点,但是没有选项指定可以保存多少个检查点。

使用simple时tf.train.Saver(),有一个选项可以指定max_to_keep

pfm*_*pfm 5

您可以将tf.train.Saver使用传递tf.train.Scaffoldtf.train.MonitoredTrainingSession

import tensorflow as tf
scaffold = tf.train.Scaffold(saver=tf.train.Saver(max_to_keep=10))
with tf.train.MonitoredTrainingSession(scaffold=scaffold) as sess:
    ...
Run Code Online (Sandbox Code Playgroud)