2017-11-26 18 views
5

का उपयोग करते समय हानि लॉगिंग संदेशों की आवृत्ति को कैसे नियंत्रित करें I TF 1.4 का उपयोग कर रहा हूं। मेरा प्रश्न tf.estimator.Estimator के बारे में है।tf.Estimator

मैं जैसे, "नुकसान और कदम" जानकारी संदेशों की आवृत्ति को नियंत्रण करना चाहते हैं:

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) 
    ]) 

उत्तर

0

मोड == 'ट्रेन' Printing extra training metrics with Tensorflow Estimator

के लिए लौट आए estimator_spec में training_hook परम रूप logging_hook लौटने की कोशिश
+2

मैंने आपके सुझाव की कोशिश की। यह मौजूदा लॉग हुक में एक नया लॉग जोड़ता है। दुर्भाग्यवश, यह मुझे डिफ़ॉल्ट लॉग हुक की आवृत्ति को प्रतिस्थापित या समायोजित करने की अनुमति नहीं देता है, जिसे बिना शर्त में जोड़ा गया है, अनुमानक में। –