INNER CODE UNIT · Python

search

neuralmagic/sparsezoo · src/sparsezoo/main.py:341

def search(args):
    LOGGER.info("loading available models...")
    models = search_models(
        domain=args.domain,
        sub_domain=args.sub_domain,
        architecture=args.architecture,
        sub_architecture=args.sub_architecture,
        framework=args.framework,
        repo=args.repo,
        dataset=args.dataset,
        training_scheme=args.training_scheme,
        sparse_name=args.sparse_name,
        sparse_category=args.sparse_category,
        sparse_target=args.sparse_target,
        release_version=args.release_version,
        page=args.page,
        page_length=args.page_length,
    )

View source record →