From 7564b7a2deaedfd09687bd9a12ddfda9b3db5261 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=AF=85?= Date: Thu, 2 Aug 2018 16:29:19 +0800 Subject: [PATCH] Fix caffe output node issue --- mace/python/tools/converter_tool/caffe_converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/python/tools/converter_tool/caffe_converter.py b/mace/python/tools/converter_tool/caffe_converter.py index 840f5dd3..5855725d 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], []), -- GitLab