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

Question

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

我的网络是使用估计器 api 实现的，我想坚持这一点，但不幸的是，我无法强制估计器跳过保存检查点。您知道一种无需保存检查点即可简单地运行一个 epoch o 训练的方法吗？

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)