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

Fix output node order

上级 e0d6d4e7
...@@ -1349,9 +1349,9 @@ class Transformer(base_converter.ConverterInterface): ...@@ -1349,9 +1349,9 @@ class Transformer(base_converter.ConverterInterface):
visited = set() visited = set()
sorted_nodes = [] sorted_nodes = []
output_nodes = self._option.check_nodes output_nodes = self._option.check_nodes.keys()
if not self._quantize_activation_info: 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: for output_node in output_nodes:
mace_check(output_node in self._producer, mace_check(output_node in self._producer,
"output_tensor %s not existed in model" % output_node) "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.
先完成此消息的编辑!
想要评论请 注册