未验证 提交 0cbf3502 编写于 作者: M mamingjie-China 提交者: GitHub

Update tf_op_mapper_nhwc.py

上级 77af5053
......@@ -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+1))
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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册