如何禁用 Tensorflow Estimator 的摘要？

Question

我在 Windows 10 上使用 Tensorflow-GPU 1.8 API。对于许多项目，我使用 tf.Estimator's，它真的很好用。它负责一系列步骤，包括为 Tensorboard 编写摘要。但是现在'events.out.tfevents'文件变得越来越大，我遇到了“空间不足”的错误。出于这个原因，我想禁用摘要编写或至少减少编写摘要的数量。

随着该任务的进行，我发现了有关 RunConfig 的信息，您可以在构建 tf.Estimator 时忽略它。显然，参数“save_summary_steps”（默认为 200）控制着摘要的输出方式。不幸的是，更改此参数似乎根本没有效果。它不会禁用（使用 None 值）摘要或减小（选择更高的值，例如 3000）'events.out.tfevents' 的文件大小。

我希望你们能在这里帮助我。任何帮助表示赞赏。

干杯， 托布斯。

Answer 1

我观察到以下行为。这对我来说没有意义，所以我希望我们能得到更好的答案：

当input_fn 从tf.data.TFRecordDataset 获取数据时，保存事件之间的步数是save_summary_steps 和（训练示例数除以batch size)。这意味着它每个 epoch 至少执行一次。

当input_fn 从tf.TextLineReader 获取数据时，它会按照您的预期跟随save_summary_steps，我可以给它一个不经常更新的较大值。