From 68e3f16923d34e2f5d8ef58e26fe6539b5401b2b Mon Sep 17 00:00:00 2001 From: liuqi Date: Thu, 7 Jun 2018 15:27:12 +0800 Subject: [PATCH] Fix validation_input_data bug. --- tools/converter.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/tools/converter.py b/tools/converter.py index b076ed4c..6d9df68d 100644 --- a/tools/converter.py +++ b/tools/converter.py @@ -371,6 +371,14 @@ def format_model_config(config_file_path): "'%s' is necessary in subgraph" % key) if not isinstance(value, list): subgraph[key] = [value] + validation_inputs_data = subgraph.get( + YAMLKeyword.validation_inputs_data, []) + if not isinstance(validation_inputs_data, list): + subgraph[YAMLKeyword.validation_inputs_data] = [ + validation_inputs_data] + else: + subgraph[YAMLKeyword.validation_inputs_data] = \ + validation_inputs_data for key in [YAMLKeyword.limit_opencl_kernel_time, YAMLKeyword.nnlib_graph_mode, @@ -380,15 +388,6 @@ def format_model_config(config_file_path): if value == "": model_config[key] = 0 - validation_inputs_data = model_config.get( - YAMLKeyword.validation_inputs_data, []) - if not isinstance(validation_inputs_data, list): - model_config[YAMLKeyword.validation_inputs_data] = [ - validation_inputs_data] - else: - model_config[YAMLKeyword.validation_inputs_data] = \ - validation_inputs_data - weight_file_path = model_config.get(YAMLKeyword.weight_file_path, "") model_config[YAMLKeyword.weight_file_path] = weight_file_path -- GitLab