提交 6fd65aae 编写于 作者: 刘琦

Merge branch 'fix_multi_inputs_converter' into 'master'

Fix multiple inputs model converter

See merge request !524
......@@ -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(',')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册