Skip to content

GitLab

Unverified Commit 6405cd2d authored by Fengzhe Zhou's avatar Fengzhe Zhou Committed by GitHub
Browse files

use example summarizer by default (#508)

parent 46e7ede9
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 7 deletions
opencompass/utils/run.py
View file @ 6405cd2d
...@@ -99,10 +99,9 @@ def get_config_from_arg(args) -> Config: ...@@ -99,10 +99,9 @@ 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, [args.summarizer])[0] s = match_cfg_file(summarizers_dir, [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']
......
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