tensorflow - 使用 tf.Estimator 时如何控制丢失日志消息的频率

Question

我正在使用 TF 1.4。我的问题是关于 tf.estimator.Estimator。

我想控制“丢失和步进”信息消息的频率，例如：

INFO:tensorflow:loss = 0.00896569, step = 14901 (14.937 sec)

我将 tf.estimator.RunConfig 传递给 Estimator 的构造函数。但我不认为有一个参数来控制“损失和步骤”消息。

我认为参数在 _train_model 方法中的 estimator.py 中是硬编码的：

      worker_hooks.extend([
      training.NanTensorHook(estimator_spec.loss),
      training.LoggingTensorHook(
          {
              'loss': estimator_spec.loss,
              'step': global_step_tensor
          },
          every_n_iter=100)
  ])