mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
15 lines
440 B
Python
15 lines
440 B
Python
|
from mmengine.config import read_base
|
||
|
|
from opencompass.openicl.icl_retriever import ZeroRetriever
|
||
|
|
from copy import deepcopy
|
||
|
|
|
||
|
|
with read_base():
|
||
|
|
from .models.qwen.hf_qwen_7b_chat import models
|
||
|
|
from .datasets.mmlu.mmlu_gen_4d595a import mmlu_datasets # this is a dataset evaluated with 5-shot
|
||
|
|
|
||
|
|
|
||
|
|
datasets = []
|
||
|
|
for d in mmlu_datasets:
|
||
|
|
d = deepcopy(d)
|
||
|
|
d['infer_cfg']['retriever'] = dict(type=ZeroRetriever)
|
||
|
|
datasets.append(d)
|