From 61c71eaaa8b8fb3de32f4ecb94f6892e1ceced20 Mon Sep 17 00:00:00 2001 From: Bin Li Date: Thu, 12 Sep 2019 17:18:28 +0800 Subject: [PATCH] Fix output node order --- tools/python/transform/transformer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/python/transform/transformer.py b/tools/python/transform/transformer.py index cc4b61f3..f4ec331a 100644 --- a/tools/python/transform/transformer.py +++ b/tools/python/transform/transformer.py @@ -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) -- GitLab