提交 61c71eaa 编写于 作者: B Bin Li 提交者: 叶剑武

Fix output node order

上级 e0d6d4e7
......@@ -1349,9 +1349,9 @@ class Transformer(base_converter.ConverterInterface):
visited = set()
sorted_nodes = []
output_nodes = self._option.check_nodes
output_nodes = self._option.check_nodes.keys()
if not self._quantize_activation_info:
output_nodes.update(self._option.output_nodes)
output_nodes.extend(self._option.output_nodes)
for output_node in output_nodes:
mace_check(output_node in self._producer,
"output_tensor %s not existed in model" % output_node)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册