如何在不保存检查点的情况下运行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)