Merge branch 'master' into 'master'

Fix caffe output node issue See merge request !703
XiaoMi · Aug 2, 2018 · b23889d · b23889d
2 parents ff3cc83 + 7564b7a
commit b23889d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mace/python/tools/converter_tool/caffe_converter.py b/mace/python/tools/converter_tool/caffe_converter.py
@@ -247,7 +247,7 @@ def replace_output_tensor_name(self):
 
         # if user set op name as output node, replace it with op name
         for op in self._mace_net_def.op:
-            if op.name in self._option.output_nodes:
+            if op.name in self._option.output_nodes and op.name not in visited:
                 if len(op.output) > 0:
                     self.replace_input_name(
                         consumers.get(op.output[0], []),