如何在不保存检查点的情况下运行estimator.train

Question

我正在寻找一种方法来实现学习率的搜索，如下所述：https://arxiv.org/pdf/1506.01186.pdf。

我的网络是使用estimator api实现的，我想坚持这一点，但不幸的是我无法强制估算器跳过保存检查点。您是否知道如何在不保存检查点的情况下简单地进行一次训练？

Answer 1

根据文档tf.estimator.RunConfig：

  

如果save_checkpoints_steps和save_checkpoints_secs都为None，则禁用检查点

所以代码如下：

run_config = tf.estimator.RunConfig(save_summary_steps=None,
                                    save_checkpoints_secs=None)
estimator = tf.estimator.Estimator(model_fn=model_fn, config=run_config)