From 13db0ab2417e72ea35c6f2b3994c366cb6e951f8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=AF=85?= Date: Thu, 24 May 2018 17:22:25 +0800 Subject: [PATCH] Fix multiple inputs model converter --- mace/python/tools/converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/python/tools/converter.py b/mace/python/tools/converter.py index 164b54e4..fe378b14 100644 --- a/mace/python/tools/converter.py +++ b/mace/python/tools/converter.py @@ -105,7 +105,7 @@ def main(unused_args): for i in xrange(len(input_node_names)): input_node = cvt.NodeInfo() input_node.name = input_node_names[i] - input_node.shape = parse_int_array_from_str(FLAGS.input_shape) + input_node.shape = parse_int_array_from_str(input_node_shapes[i]) option.add_input_node(input_node) output_node_names = FLAGS.output_node.split(',') -- GitLab