chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 02:20: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

9
minigpt4/common/config.py
View File

@ -116,13 +116,14 @@ 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()
if datasets is not None:
for dataset_name in datasets: for dataset_name in datasets:
builder_cls = registry.get_builder_class(dataset_name) builder_cls = registry.get_builder_class(dataset_name)