refine transpose optimizer's log

Author: zhijxu-MS

tf2onnx/optimizer/optimizer_base.py

@@ -52,4 +52,4 @@ def _print_stat_diff(self, nodes_original, nodes_after_optimized):
         for key, value in nodes_after_optimized.items():
             if value != 0:
                 res[key] = value
-        self._log.info("after optimized, the optimization_statistics is %s", res)
+        self._log.info("the optimization gain is %s", res)

tf2onnx/optimizer/transpose_optimizer.py

@@ -13,9 +13,6 @@
 from tf2onnx import utils
 from tf2onnx.optimizer.optimizer_base import GraphOptimizerBase
 
-logging.basicConfig(level=logging.INFO)
-log = logging.getLogger("tf2onnx.optimizer.transpose_optimizer")
-
 
 # pylint: disable=logging-not-lazy,unused-argument,missing-docstring,abstract-method
 # FIXME:
@@ -41,6 +38,7 @@ class TransposeOptimizer(GraphOptimizerBase):
 
     def __init__(self, debug=False):
         super(TransposeOptimizer, self).__init__("TransposeOptimizer", debug)
+        self._log = logging.getLogger("tf2onnx.optimizer.%s" % self._name)
         self._handler_map = {}
         self._force_stop = {}
 
@@ -158,17 +156,17 @@ def optimize(self, graph):
             if "stop" in self._force_stop and self._force_stop["stop"] == 1:
                 break
 
-        log.debug("finish after " + str(iteration_cnt) + " iteration(s)")
+        self._log.debug("finish after " + str(iteration_cnt) + " iteration(s)")
 
         self.merge_duplicated_transposes()
         self.post_optimize_action()
 
         current_counter = self._g.dump_node_statistics()
         transpose_cnt = current_counter["Transpose"]
-        current_counter.subtract(previous_counter)
-        log.info(" %d transpose op(s) left, ops diff after transpose optimization: %s", transpose_cnt, current_counter)
+        self._log.info(" %d transpose op(s) left", transpose_cnt)
+        self._print_stat_diff(previous_counter, current_counter)
         if transpose_cnt > 2:
-            log.warning("please try add --fold_const to help remove more transpose")
+            self._log.warning("please try add --fold_const to help remove more transpose")
         return self._g
 
     def _initialize_handlers(self):
@@ -219,7 +217,7 @@ def _handle_node_having_branches(self, node):
                 self._g.remove_node(n.name)
             return True
 
-        log.debug("input transpose does not have single consumer, skipping...")
+        self._log.debug("input transpose does not have single consumer, skipping...")
         return False
 
     # get the input index of transpose op in node's inputs.
@@ -263,7 +261,7 @@ def _handle_nhwc_tranpose(self, trans):
         if len(out_nodes) == 1:
             p = out_nodes[0]
             if p.name in self._output_names:
-                log.debug("cannot move transpose down since it met output node %s", p.name)
+                self._log.debug("cannot move transpose down since it met output node %s", p.name)
                 return False
 
             if p.type in self._handler_map: