From 6d3eeebe798be4533e866e85746cfce7084b1c4a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=8F=B6=E5=89=91=E6=AD=A6?= Date: Mon, 11 Nov 2019 10:11:49 +0800 Subject: [PATCH] support int types for inputs/outputs name format code --- tools/python/utils/config_parser.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/python/utils/config_parser.py b/tools/python/utils/config_parser.py index 56b7a341..9e5c9f6d 100644 --- a/tools/python/utils/config_parser.py +++ b/tools/python/utils/config_parser.py @@ -215,6 +215,8 @@ def normalize_model_config(conf): # parse input conf[ModelKeys.input_tensors] = to_list(conf[ModelKeys.input_tensors]) + conf[ModelKeys.input_tensors] = [str(i) for i in + conf[ModelKeys.input_tensors]] input_count = len(conf[ModelKeys.input_tensors]) conf[ModelKeys.input_shapes] = [parse_int_array(shape) for shape in to_list(conf[ModelKeys.input_shapes])] @@ -255,6 +257,8 @@ def normalize_model_config(conf): # parse output conf[ModelKeys.output_tensors] = to_list(conf[ModelKeys.output_tensors]) + conf[ModelKeys.output_tensors] = [str(i) for i in + conf[ModelKeys.output_tensors]] output_count = len(conf[ModelKeys.output_tensors]) conf[ModelKeys.output_shapes] = [parse_int_array(shape) for shape in to_list(conf[ModelKeys.output_shapes])] -- GitLab