jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

use example summarizer by default (#508)

Browse Source
This commit is contained in:
Fengzhe Zhou 2023-10-27 11:45:29 +08:00 committed by GitHub
parent 46e7ede938
commit 6405cd2db5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
opencompass/utils/run.py
View File

@ -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),