This commit is contained in:
zhangsongyang 2025-05-06 09:55:26 +00:00
parent 61a2a6b763
commit ba0186ba1c
3 changed files with 12 additions and 5 deletions

View File

@ -112,6 +112,7 @@ livemathbench_datasets = [
judge_cfg={},
dict_postprocessor=dict(type=generic_llmjudge_postprocess),
),
parallel=False
),
),
n=1, # repeat n times

View File

@ -110,6 +110,7 @@ for name in mmlu_all_sets:
dict_postprocessor=dict(type=generic_llmjudge_postprocess),
judge_cfg=dict(),
),
parallel=False
),
)

View File

@ -146,11 +146,16 @@ class OpenICLEvalTask(BaseTask):
preds = []
i = 1
while osp.exists(osp.realpath(filename)):
sub_preds = mmengine.load(filename)
preds.extend(
[sub_preds[str(i)] for i in range(len(sub_preds))])
filename = root + f'_{i}' + ext
i += 1
try:
sub_preds = mmengine.load(filename)
preds.extend(
[sub_preds[str(i)] for i in range(len(sub_preds))])
filename = root + f'_{i}' + ext
i += 1
except Exception as e:
self.logger.error(
f'Error loading prediction file {filename}: {e}')
break
pred_dicts = copy.deepcopy(preds)
preds = {k: [pred.get(k) for pred in preds] for k in preds[0]}