Skip to content

GitLab

"git@developer.sourcefind.cn:yangql/composable_kernel-1.git" did not exist on "4f0fc72e913965d92d5410eb2a1c4a0e6b1fce6f"
Unverified Commit 6405cd2d authored by Fengzhe Zhou's avatar Fengzhe Zhou Committed by GitHub
Browse files

use example summarizer by default (#508)

parent 46e7ede9
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 7 deletions
opencompass/utils/run.py
View file @ 6405cd2d
...@@ -99,13 +99,12 @@ def get_config_from_arg(args) -> Config: ...@@ -99,13 +99,12 @@ def get_config_from_arg(args) -> Config:
run_cfg=dict(num_gpus=args.num_gpus)) run_cfg=dict(num_gpus=args.num_gpus))
models.append(model) models.append(model)
summarizer = {} summarizer = args.summarizer if args.summarizer is not None else 'example'
if args.summarizer: summarizers_dir = os.path.join(args.config_dir, 'summarizers')
summarizers_dir = os.path.join(args.config_dir, 'summarizers') s = match_cfg_file(summarizers_dir, [summarizer])[0]
s = match_cfg_file(summarizers_dir, [args.summarizer])[0] get_logger().info(f'Loading {s[0]}: {s[1]}')
get_logger().info(f'Loading {s[0]}: {s[1]}') cfg = Config.fromfile(s[1])
cfg = Config.fromfile(s[1]) summarizer = cfg['summarizer']
summarizer = cfg['summarizer']
return Config(dict(models=models, datasets=datasets, return Config(dict(models=models, datasets=datasets,
summarizer=summarizer), summarizer=summarizer),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment