eval_demo.py 342 Bytes
Newer Older
Leymore's avatar
Leymore committed
1
2
from mmengine.config import read_base

yuzhaohui's avatar
yuzhaohui committed
3
with read_base():
Ezra-Yu's avatar
Ezra-Yu committed
4
    from .datasets.siqa.siqa_gen import siqa_datasets
5
6
7
    from .datasets.winograd.winograd_ppl import winograd_datasets
    from .models.hf_opt_125m import opt125m
    from .models.hf_opt_350m import opt350m
Ezra-Yu's avatar
Ezra-Yu committed
8
9

datasets = [*siqa_datasets, *winograd_datasets]
10
models = [opt125m, opt350m]