INNER CODE UNIT · Python

main

neuralmagic/sparsezoo · src/sparsezoo/deployment_package/cli.py:182

def main(**kwargs):
    r"""
    Utility to fetch a deployment directory for a task based on specified
    optimizing-metric

    Example for using sparsezoo.deployment_package:

         1) `sparsezoo.deployment_package --task image_classification \
                --optimizing_metric accuracy

         2) `sparsezoo.deployment_package --task ic \
                --optimizing_metric "accuracy, compression" \
                --target VNNI`
    """
    if not (kwargs.get("task") or kwargs.get("dataset")):
        raise ValueError("At-least one of the `task` or `dataset` must be specified")
    _LOGGER.debug(f"{kwargs}")
    results = deployment_package(**kwargs)

View source record →