diff --git a/tools/python/utils/config_parser.py b/tools/python/utils/config_parser.py index 56b7a341612fe9931221ede7afe235d50334ccd6..9e5c9f6dc843e8bb9a54f1e65fc99fd75ab1b455 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])]