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

Question

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

我想控制“损失和步伐”信息消息的出现频率，例如:

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

我正在将tf.estimator.RunConfig传递给Estimator的构造函数。但我认为没有参数可以控制“损失和损失”消息。

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

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

Answer 1

tensorflow v1.8支持log_step_count_steps:https://www.tensorflow.org/api_docs/python/tf/estimator/RunConfig