[Fix] fix mim search error (#3194)

This commit is contained in:
谢昕辰 2023-07-13 16:26:08 +08:00 committed by GitHub
parent ac78b1308d
commit 067a95e40b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 11 additions and 17 deletions

View File

@ -78,6 +78,9 @@ def get_model_info(md_file: str, config_dir: str,
if line.startswith('<!-- [DATASET]'):
is_dataset = True
if '<!-- [SKIP DEV CHECK] -->' in line:
return None, None
# get dataset names
if line.startswith('###'):
current_dataset = line.split('###')[1].strip()
@ -279,14 +282,20 @@ if __name__ == '__main__':
collection_name_list: List[str] = get_collection_name_list(md_file_list)
# hard code to add 'FPN'
collection_name_list.append('FPN')
remove_config_dir_list = []
# parse md file
for md_file, config_dir in zip(md_file_list, config_dir_list):
results, collection_name = get_model_info(md_file, config_dir,
collection_name_list)
if results is None:
remove_config_dir_list.append(config_dir)
continue
filename = osp.join(config_dir, 'metafile.yaml')
file_modified |= dump_yaml_and_check_difference(results, filename)
if collection_name != '':
collection_name_list.append(collection_name)
# remove config dir
for config_dir in remove_config_dir_list:
config_dir_list.remove(config_dir)
file_modified |= update_model_index(config_dir_list)
sys.exit(1 if file_modified else 0)

View File

@ -1,8 +1,6 @@
# DSDL: Standard Description Language for DataSet
<!-- [ALGORITHM] -->
<!-- [DATASET] -->
<!-- [SKIP DEV CHECK] -->
## Abstract

View File

@ -1,12 +0,0 @@
Collections:
- Name: ''
License: Apache License 2.0
Metadata:
Training Data: []
Paper:
Title: ''
URL: ''
README: configs/dsdl/README.md
Frameworks:
- PyTorch
Models: []

View File

@ -14,7 +14,6 @@ Import:
- configs/dmnet/metafile.yaml
- configs/dnlnet/metafile.yaml
- configs/dpt/metafile.yaml
- configs/dsdl/metafile.yaml
- configs/emanet/metafile.yaml
- configs/encnet/metafile.yaml
- configs/erfnet/metafile.yaml