gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 18:21:28 39 4
gpt4 key购买 nike

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

我的网络是使用 estimator api 实现的,我想坚持下去,但不幸的是,我无法强制 estimator 跳过保存检查点。您知道一种无需保存检查点就可以简单地运行一个 epoch o 训练的方法吗?

最佳答案

根据文档 tf.estimator.RunConfig :

If both save_checkpoints_steps and save_checkpoints_secs are None, then checkpoints are disabled



所以代码如下:
run_config = tf.estimator.RunConfig(save_summary_steps=None,
save_checkpoints_secs=None)
estimator = tf.estimator.Estimator(model_fn=model_fn, config=run_config)

关于tensorflow - 如何在不保存检查点的情况下运行 estimator.train,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48361703/

39 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com