Cách kiểm soát tần suất thông báo ghi nhật ký mất mát khi sử dụng tf.Estimator

Question

Tôi đang sử dụng TF 1.4. Câu hỏi của tôi là về tf.estimator.Estimator.

Tôi muốn kiểm soát tần số của "mất mát và bước" Tin thông báo, như:

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

Tôi đang đi qua một tf.estimator.RunConfig đến constructor của Estimator. Nhưng tôi không nghĩ rằng có một tham số để kiểm soát các thông báo "mất và bước".

Tôi nghĩ rằng các tham số được mã hóa cứng trong estimator.py, trong phương pháp _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

thử trả lại logging_hook như training_hook param trong estimator_spec trở cho chế độ == 'tàu' Printing extra training metrics with Tensorflow Estimator