提交 b23889d5 编写于 作者: 刘琦

Merge branch 'master' into 'master'

Fix caffe output node issue

See merge request !703
......@@ -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], []),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册