使用时tf.train.MonitoredTrainingSession
,可以保存所有检查点。它有一个参数(save_checkpoint_secs=600
)来指定要保存多少个检查点,但是没有选项指定可以保存多少个检查点。
使用simple时tf.train.Saver()
,有一个选项可以指定max_to_keep
。
您可以将tf.train.Saver
使用传递tf.train.Scaffold
给tf.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)
归档时间: |
|
查看次数: |
1245 次 |
最近记录: |