我正在寻找一种方法来实现学习率搜索,如下所述:https://arxiv.org/pdf/1506.01186.pdf https://arxiv.org/pdf/1506.01186.pdf .
我的网络是使用估算器 api 实现的,我想坚持这一点,但不幸的是我无法强制估算器跳过保存检查点。您是否知道一种简单地运行一个周期训练而不保存检查点的方法?
根据文档tf.estimator.RunConfig https://www.tensorflow.org/api_docs/python/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)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)