OpenCompass/examples/eval_mmlu_with_zero_retriever_overwritten.py

17 lines
490 B
Python
Raw Normal View History

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)