From 6405cd2db586a6246f806f27d2302ed533209101 Mon Sep 17 00:00:00 2001 From: Fengzhe Zhou Date: Fri, 27 Oct 2023 11:45:29 +0800 Subject: [PATCH] use example summarizer by default (#508) --- opencompass/utils/run.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/opencompass/utils/run.py b/opencompass/utils/run.py index 10f58549..0c5b1c3f 100644 --- a/opencompass/utils/run.py +++ b/opencompass/utils/run.py @@ -99,13 +99,12 @@ def get_config_from_arg(args) -> Config: run_cfg=dict(num_gpus=args.num_gpus)) models.append(model) - summarizer = {} - if args.summarizer: - summarizers_dir = os.path.join(args.config_dir, 'summarizers') - s = match_cfg_file(summarizers_dir, [args.summarizer])[0] - get_logger().info(f'Loading {s[0]}: {s[1]}') - cfg = Config.fromfile(s[1]) - summarizer = cfg['summarizer'] + summarizer = args.summarizer if args.summarizer is not None else 'example' + summarizers_dir = os.path.join(args.config_dir, 'summarizers') + s = match_cfg_file(summarizers_dir, [summarizer])[0] + get_logger().info(f'Loading {s[0]}: {s[1]}') + cfg = Config.fromfile(s[1]) + summarizer = cfg['summarizer'] return Config(dict(models=models, datasets=datasets, summarizer=summarizer),