"include/vscode:/vscode.git/clone" did not exist on "d714fa15cb556fcb5f132f1a0c7cfbc22163a253"
Unverified Commit e257254b authored by hailsham's avatar hailsham Committed by GitHub
Browse files

[Feature] add global retriever config (#842)



* add global retriever config

* give zero shot overwrite example

* give zero shot overwrite example

---------
Co-authored-by: default avatarLei Fei <SENSETIME\leifei1@cn3114002087l.domain.sensetime.com>
Co-authored-by: default avatarLeymore <zfz-960727@163.com>
parent dd444685
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)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment