This commit is contained in:
zhangsongyang 2025-05-14 10:14:52 +00:00
parent 6ea181d15c
commit cf7f2425cf

View File

@ -156,6 +156,7 @@ def get_config_from_arg(args) -> Config:
for _dataset in datasets:
logger.warning(f"The default num runs of {_dataset['abbr']} is: {_dataset['n']}, changed into: {args.dataset_num_runs}")
_dataset['n'] = args.dataset_num_runs
_dataset['k'] = args.dataset_num_runs
# parse model args
if not args.models and not args.hf_path: