mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
[Sync] bump version 0.2.6+local (#1294)
This commit is contained in:
parent
1d3a26c732
commit
a62c613d3e
@ -1 +1 @@
|
|||||||
__version__ = '0.2.5'
|
__version__ = '0.2.6'
|
||||||
|
@ -4,7 +4,6 @@ import argparse
|
|||||||
import getpass
|
import getpass
|
||||||
import os
|
import os
|
||||||
import os.path as osp
|
import os.path as osp
|
||||||
from copy import deepcopy
|
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
from mmengine.config import Config, DictAction
|
from mmengine.config import Config, DictAction
|
||||||
@ -349,7 +348,7 @@ def main():
|
|||||||
|
|
||||||
# For subjective summarizer
|
# For subjective summarizer
|
||||||
if summarizer_cfg.get('function', None):
|
if summarizer_cfg.get('function', None):
|
||||||
main_summarizer_cfg = deepcopy(summarizer_cfg)
|
main_summarizer_cfg = copy.deepcopy(summarizer_cfg)
|
||||||
grouped_datasets = {}
|
grouped_datasets = {}
|
||||||
for dataset in cfg.datasets:
|
for dataset in cfg.datasets:
|
||||||
prefix = dataset['abbr'].split('_')[0]
|
prefix = dataset['abbr'].split('_')[0]
|
||||||
@ -361,7 +360,7 @@ def main():
|
|||||||
all_grouped_lists.append(grouped_datasets[prefix])
|
all_grouped_lists.append(grouped_datasets[prefix])
|
||||||
dataset_score_container = []
|
dataset_score_container = []
|
||||||
for dataset in all_grouped_lists:
|
for dataset in all_grouped_lists:
|
||||||
temp_cfg = deepcopy(cfg)
|
temp_cfg = copy.deepcopy(cfg)
|
||||||
temp_cfg.datasets = dataset
|
temp_cfg.datasets = dataset
|
||||||
summarizer_cfg = dict(type=dataset[0]['summarizer']['type'], config=temp_cfg)
|
summarizer_cfg = dict(type=dataset[0]['summarizer']['type'], config=temp_cfg)
|
||||||
summarizer = build_from_cfg(summarizer_cfg)
|
summarizer = build_from_cfg(summarizer_cfg)
|
||||||
@ -378,5 +377,7 @@ def main():
|
|||||||
summarizer = build_from_cfg(summarizer_cfg)
|
summarizer = build_from_cfg(summarizer_cfg)
|
||||||
summarizer.summarize(time_str=cfg_time_str)
|
summarizer.summarize(time_str=cfg_time_str)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
main()
|
main()
|
||||||
|
@ -4,7 +4,6 @@ import os.path as osp
|
|||||||
import re
|
import re
|
||||||
|
|
||||||
from datasets import Dataset, DatasetDict
|
from datasets import Dataset, DatasetDict
|
||||||
from torch.utils.data import DataLoader
|
|
||||||
|
|
||||||
from opencompass.registry import LOAD_DATASET
|
from opencompass.registry import LOAD_DATASET
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user