mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
Update
This commit is contained in:
parent
61a2a6b763
commit
ba0186ba1c
@ -112,6 +112,7 @@ livemathbench_datasets = [
|
||||
judge_cfg={},
|
||||
dict_postprocessor=dict(type=generic_llmjudge_postprocess),
|
||||
),
|
||||
parallel=False
|
||||
),
|
||||
),
|
||||
n=1, # repeat n times
|
||||
|
@ -110,6 +110,7 @@ for name in mmlu_all_sets:
|
||||
dict_postprocessor=dict(type=generic_llmjudge_postprocess),
|
||||
judge_cfg=dict(),
|
||||
),
|
||||
parallel=False
|
||||
),
|
||||
)
|
||||
|
||||
|
@ -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]}
|
||||
|
Loading…
Reference in New Issue
Block a user