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

[Enhancement] Increase default task size (#360)

Browse Source
This commit is contained in:
Tong Gao 2023-09-05 10:38:13 +08:00 committed by GitHub
parent b8bf16e81c
commit 5d75c1bbb9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
opencompass/partitioners/size.py
View File

@ -29,7 +29,7 @@ class SizePartitioner(BasePartitioner):
def __init__(self,
out_dir: str,
max_task_size: int = 2000,
max_task_size: int = 40000,
gen_task_coef: int = 20,
dataset_size_path: str = '.cache/dataset_size.json'):
super().__init__(out_dir)

2
run.py
View File

@ -91,7 +91,7 @@ def parse_args():
help='The maximum size of an infer task. Only '
'effective when "infer" is missing from the config.',
type=int,
default=2000),
default=40000),
parser.add_argument(
'--gen-task-coef',
help='The dataset cost measurement coefficient for generation tasks, '