chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-04 18:10:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #407 from junchen14/main

Browse Source 
update evaluation configs
This commit is contained in:
Jun Chen 2023-11-01 16:12:45 -07:00 committed by GitHub
commit 353bcc7e3a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
minigpt4/common/config.py
View File

@ -116,21 +116,22 @@ class Config:
@staticmethod @staticmethod
def build_evaluation_dataset_config(config): def build_evaluation_dataset_config(config):
datasets = config.get("evaluation_datasets", None) datasets = config.get("evaluation_datasets", None)
if datasets is None: # if datasets is None:
raise KeyError( # raise KeyError(
"Expecting 'datasets' as the root key for dataset configuration." # "Expecting 'datasets' as the root key for dataset configuration."
) # )
dataset_config = OmegaConf.create() dataset_config = OmegaConf.create()
for dataset_name in datasets: if datasets is not None:
builder_cls = registry.get_builder_class(dataset_name) for dataset_name in datasets:
builder_cls = registry.get_builder_class(dataset_name)
# hierarchy override, customized config > default config # hierarchy override, customized config > default config
dataset_config = OmegaConf.merge( dataset_config = OmegaConf.merge(
dataset_config, dataset_config,
{"evaluation_datasets": {dataset_name: config["evaluation_datasets"][dataset_name]}}, {"evaluation_datasets": {dataset_name: config["evaluation_datasets"][dataset_name]}},
) )
return dataset_config return dataset_config