From 8542e29f37ca757635542d169f1ac3016f808fae Mon Sep 17 00:00:00 2001 From: lichangwei Date: Wed, 6 Aug 2025 16:13:20 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90bugfix=E3=80=91config=5Fcheck=20yaml?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E8=A7=A3=E6=9E=90=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../core/config_check/utils/hyperparameter_parser.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/debug/accuracy_tools/msprobe/core/config_check/utils/hyperparameter_parser.py b/debug/accuracy_tools/msprobe/core/config_check/utils/hyperparameter_parser.py index 6cb540ee49..a524504c47 100644 --- a/debug/accuracy_tools/msprobe/core/config_check/utils/hyperparameter_parser.py +++ b/debug/accuracy_tools/msprobe/core/config_check/utils/hyperparameter_parser.py @@ -96,9 +96,13 @@ class YamlParser(Parser): new_prefix = prefix + Const.SEP + key if prefix else key self.recursive_parse_parameters(value, new_prefix) elif isinstance(parameters, list): - for value in parameters: - self.recursive_parse_parameters(value, prefix) - elif isinstance(parameters, (int, str, bool)): + if all(isinstance(x, (int, float, str, bool))for x in parameters): + self.hyperparameters.update({prefix: parameters}) + else: + for idx, value in enumerate(parameters): + new_prefix = prefix + Const.SEP + str(idx) if prefix else str(idx) + self.recursive_parse_parameters(value, new_prefix) + elif isinstance(parameters, (int, float, str, bool)): self.hyperparameters.update({prefix: parameters}) -- Gitee