Explorar o código

Set tf.logging verbosity to INFO

Neal Wu %!s(int64=8) %!d(string=hai) anos
pai
achega
844c1eb4c2
Modificáronse 2 ficheiros con 4 adicións e 5 borrados
  1. 3 4
      resnet/resnet_main.py
  2. 1 1
      resnet/resnet_model.py

+ 3 - 4
resnet/resnet_main.py

@@ -162,10 +162,8 @@ def evaluate(hps):
         tag='Best Precision', simple_value=best_precision)
     summary_writer.add_summary(best_precision_summ, train_step)
     summary_writer.add_summary(summaries, train_step)
-    output_str = 'loss: %.3f, precision: %.3f, best precision: %.3f\n' % \
-                 (loss, precision, best_precision)
-    tf.logging.info(output_str)
-    print(output_str)
+    tf.logging.info('loss: %.3f, precision: %.3f, best precision: %.3f' %
+                    (loss, precision, best_precision))
     summary_writer.flush()
 
     if FLAGS.eval_once:
@@ -210,4 +208,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.logging.set_verbosity(tf.logging.INFO)
   tf.app.run()

+ 1 - 1
resnet/resnet_model.py

@@ -221,7 +221,7 @@ class ResNet(object):
                      [(out_filter-in_filter)//2, (out_filter-in_filter)//2]])
       x += orig_x
 
-    tf.logging.info('image after unit %s', x.get_shape())
+    tf.logging.debug('image after unit %s', x.get_shape())
     return x
 
   def _bottleneck_residual(self, x, in_filter, out_filter, stride,