如何控制张量流量估算器保存的检查点数量？

Question

我注意到新的Estimator API会在训练期间自动保存检查点,并在训练中断时自动从上一个检查点重新启动.不幸的是,它似乎只保留了最后5个检查点.

您知道如何控制培训期间保留的检查点数量吗？

Answer 1

Tensorflow tf.estimator.Estimator需要config作为可选参数,其可以是tf.estimator.RunConfig对象配置运行时settings.You可以实现这样的,如下所示:

# Change maximum number checkpoints to 25
run_config = tf.estimator.RunConfig()
run_config = run_config.replace(keep_checkpoint_max=25)

# Build your estimator
estimator = tf.estimator.Estimator(model_fn,
                                   model_dir=job_dir,
                                   config=run_config,
                                   params=None)

config参数是在所有类(得DNNClassifier,DNNLinearCombinedClassifier,LinearClassifier延伸等)estimator.Estimator.