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

update config files

Browse Source
This commit is contained in:
junchen14 2023-11-02 02:07:07 +03:00
parent 748d18b5df
commit d057192f67
1 changed files with 12 additions and 11 deletions

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