Как контролировать частоту сообщений о регистрации потерь при использовании tf.Estimator

Я использую TF 1.4. У меня вопрос по поводу тф.этиматор.Эстиматор.

Я хотел бы контролировать частоту информационных сообщений "Потеря и шаг", например:

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

Я передаю tf.estimator.RunConfig в конструктор Оценщика. Но я не думаю, что есть параметр для управления сообщениями "Потеря и шаг".

Я думаю, что параметр жестко запрограммирован в 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)
  ])

2 ответа

log_step_count_steps поддерживается в tenorflow v1.8: https://www.tensorflow.org/api_docs/python/tf/estimator/RunConfig

Другие вопросы по тегам