From 77af5053b5b12246cc006ac692d02bb4f5768008 Mon Sep 17 00:00:00 2001 From: mamingjie-China <52770436+mamingjie-China@users.noreply.github.com> Date: Wed, 11 Sep 2019 19:18:07 +0800 Subject: [PATCH] Update tf_op_mapper_nhwc.py --- x2paddle/op_mapper/tf_op_mapper_nhwc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/x2paddle/op_mapper/tf_op_mapper_nhwc.py b/x2paddle/op_mapper/tf_op_mapper_nhwc.py index 9399c6d..01fd539 100644 --- a/x2paddle/op_mapper/tf_op_mapper_nhwc.py +++ b/x2paddle/op_mapper/tf_op_mapper_nhwc.py @@ -74,7 +74,7 @@ class TFOpMapperNHWC(OpMapper): unsupported_ops = set() sys.stderr.write("Total nodes: {}\n".format(len(self.graph.topo_sort))) for i, node_name in enumerate(self.graph.topo_sort): - sys.stderr.write("\rConverting node {} ... ".format(i)) + sys.stderr.write("\rConverting node {} ... ".format(i+1)) node = self.graph.get_node(node_name) op = node.layer_type if op in self.directly_map_ops: @@ -99,7 +99,7 @@ class TFOpMapperNHWC(OpMapper): for op in unsupported_ops: print("========== {} ============".format(op)) sys.exit(-1) - sys.stderr.write("\nDone\n") + sys.stderr.write("\nDone!\n") def add_omit_nodes(self, in_node_name, out_node_name): in_node = self.graph.get_node(in_node_name) -- GitLab