如何控制tensorflow估计器保留的检查点数量?

如何控制tensorflow估计器保留的检查点数量?,tensorflow,tensorflow-estimator,Tensorflow,Tensorflow Estimator,我注意到,新的Estimator API在培训期间自动保存检查点,并在培训中断时自动从最后一个检查点重新启动。不幸的是,它似乎只保留最后5个检查点 您知道如何控制培训期间保留的检查点数量吗?Tensorflow将config作为可选参数,可以作为配置运行时设置的对象。您可以通过以下方式实现此目的: #将最大检查点数更改为25 run_config=tf.estimator.RunConfig() run\u config=run\u config.replace(保留检查点\u max=25)

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

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

Tensorflow将
config
作为可选参数,可以作为配置运行时设置的对象。您可以通过以下方式实现此目的:

#将最大检查点数更改为25
run_config=tf.estimator.RunConfig()
run\u config=run\u config.replace(保留检查点\u max=25)
#建立你的估计器
估计器=tf.估计器.估计器(模型_fn,
model_dir=job_dir,
config=run\u config,
参数=无)

config
参数在所有类(
DNNClassifier
DNNLinearCombinedClassifier
LinearClassifier
等)中都可用,这些类扩展了
estimator.estimator

所需的准确信息,而且RunConfig还有额外的参数,比如
save\u checkpoints\u secs
save\u checkpoints\u steps
,完美!非常感谢。