5
TF 1.4를 사용하고 있습니다. 제 질문은 tf.estimator.Estimator에 관한 것입니다.tf.Estimator를 사용할 때 손실 로깅 메시지의 빈도를 제어하는 방법
내가처럼 "손실 단계"정보 메시지의 주파수를 제어하고 싶습니다:
INFO:tensorflow:loss = 0.00896569, step = 14901 (14.937 sec)
내가 견적의 생성자에 tf.estimator.RunConfig을 전달하고있다. 그러나 "손실 및 단계"메시지를 제어하는 매개 변수는 없다고 생각합니다.
worker_hooks.extend([
training.NanTensorHook(estimator_spec.loss),
training.LoggingTensorHook(
{
'loss': estimator_spec.loss,
'step': global_step_tensor
},
every_n_iter=100)
])
나는 당신의 제안을 시도했다. 기존 로그 후크에 새 로그를 추가합니다. 유감스럽게도, estimator.py에서 무조건 추가되는 기본 로그 후크의 빈도를 바꾸거나 조정할 수 없습니다. –