mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
34 lines
988 B
Python
34 lines
988 B
Python
|
import argparse
|
||
|
|
|
||
|
|
import tabulate
|
||
|
|
|
||
|
|
from opencompass.utils import match_files
|
||
|
|
|
||
|
|
|
||
|
|
def parse_args():
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
description='Utils to list available models and datasets.')
|
||
|
|
parser.add_argument('pattern',
|
||
|
|
nargs='*',
|
||
|
|
default='*',
|
||
|
|
type=str,
|
||
|
|
help='Patterns, '
|
||
|
|
'wildcard matching supported.')
|
||
|
|
return parser.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
args = parse_args()
|
||
|
|
models = match_files('configs/models/', args.pattern, fuzzy=True)
|
||
|
|
if models:
|
||
|
|
table = [['Model', 'Config Path'], *models]
|
||
|
|
print(tabulate.tabulate(table, headers='firstrow', tablefmt='psql'))
|
||
|
|
datasets = match_files('configs/datasets/', args.pattern, fuzzy=True)
|
||
|
|
if datasets:
|
||
|
|
table = [['Dataset', 'Config Path'], *datasets]
|
||
|
|
print(tabulate.tabulate(table, headers='firstrow', tablefmt='psql'))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|