小编Mic*_*ein的帖子

我正在使用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)
  ])