diff --git a/mace/python/tools/converter_tool/caffe_converter.py b/mace/python/tools/converter_tool/caffe_converter.py index 840f5dd3ec49b26d811a11a5b61d115393974f26..5855725d8a552aec4bd5d0de3a9955ccb4743d07 100644 --- a/mace/python/tools/converter_tool/caffe_converter.py +++ b/mace/python/tools/converter_tool/caffe_converter.py @@ -247,7 +247,7 @@ class CaffeConverter(base_converter.ConverterInterface): # 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], []),