eval_mmlu_with_zero_retriever_overwritten.py 490 Bytes
Newer Older
jerrrrry's avatar
jerrrrry committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from copy import deepcopy

from mmengine.config import read_base

from opencompass.openicl.icl_retriever import ZeroRetriever

with read_base():
    from opencompass.configs.datasets.mmlu.mmlu_gen_4d595a import \
        mmlu_datasets  # this is a dataset evaluated with 5-shot
    from opencompass.configs.models.qwen.hf_qwen_7b_chat import models

datasets = []
for d in mmlu_datasets:
    d = deepcopy(d)
    d['infer_cfg']['retriever'] = dict(type=ZeroRetriever)
    datasets.append(d)