Fix multiple inputs model converter

13db0ab2 · 李寅 · d6951a4c · 13db0ab2
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

mace/python/tools/converter.py mace/python/tools/converter.py +1 -1

未找到文件。
--- 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(',')