refine logging

Author: nbcsm

tf2onnx/optimizer/__init__.py

@@ -50,7 +50,7 @@ def optimize_graph(graph):
     diff = copy.deepcopy(after)
     diff.subtract(before)
     diff = ["{} {} ({}->{})".format(k, str(v) if v < 0 else '+' + str(v), before.get(k, 0), after.get(k, 0))
-            for k, v in diff.most_common() if v != 0]
+            for k, v in sorted(diff.items()) if v != 0]
     logger.info("After optimization: %s", ', '.join(diff) if diff else "no change")
 
     return graph

tf2onnx/optimizer/const_fold_optimizer.py

@@ -69,7 +69,7 @@ def _fold_node(self, node, graph):
                 const_outputs = process_func(node, graph)
                 self._replace_node_with_const(node, graph, const_outputs)
                 return True
-        self.logger.debug("need to add function to fold op %s whose op_type is %s", node.name, node.type)
+            self.logger.debug("need to add function to fold op %s whose op_type is %s", node.name, node.type)
         return False
 
     @staticmethod

tf2onnx/optimizer/optimizer_base.py

@@ -63,5 +63,5 @@ def _print_stat_diff(self, before, after):
         diff = copy.deepcopy(after)
         diff.subtract(before)
         diff = ["{} {} ({}->{})".format(k, str(v) if v < 0 else '+' + str(v), before.get(k, 0), after.get(k, 0))
-                for k, v in diff.most_common() if v != 0]
+                for k, v in sorted(diff.items()) if v != 0]
         self.logger.verbose(', '.join(diff) if diff else "no change")

tf2onnx/shape_inference.py

@@ -24,7 +24,9 @@
 def infer_shape(tf_graph, shape_override):
     """Infer shape for TF graph with shape_override set first."""
     if shape_override:
+        logger.info("Apply shape override:")
         for name, shape in shape_override.items():
+            logger.info("\tSet %s shape to %s", name, shape)
             tf_graph.get_tensor_by_name(name).set_shape(shape)
         tf_graph = reload_tf_graph(tf_graph)