From 5d75c1bbb93618942abdf1e198c97d4fe6fed671 Mon Sep 17 00:00:00 2001
From: Tong Gao <gaotongxiao@gmail.com>
Date: Tue, 5 Sep 2023 10:38:13 +0800
Subject: [PATCH] [Enhancement] Increase default task size (#360)

---
 opencompass/partitioners/size.py | 2 +-
 run.py                           | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/opencompass/partitioners/size.py b/opencompass/partitioners/size.py
index 9b7f8c6a..7b94f915 100644
--- a/opencompass/partitioners/size.py
+++ b/opencompass/partitioners/size.py
@@ -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)
diff --git a/run.py b/run.py
index 3c3c5d1c..b52a664a 100644
--- a/run.py
+++ b/run.py
@@ -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, '