Python export_model示例

示例#1

def export(name, framework, trt):
    """
    Export model from PyTorch hub / TensorFlow hub and try convert the model into various format for different serving
    engines.
    """
    export_model(model_name=name, framework=framework, enable_trt=trt)
    exit(0)

示例#2

文件： modelhub.py 项目： univerone/ML-Model-CI

def export(
        name: str = typer.Option(..., '-n', '--name', help='Architecture'),
        framework: Optional[Framework] = typer.Option(None, '-fw', '--framework', case_sensitive=False,
                                                      help='Framework'),
        trt: Optional[bool] = typer.Option(
            False,
            is_flag=True,
            help='Flag for exporting models served by TensorRT. Please make sure you have TensorRT installed in your '
                 'machine before set this flag.')
):
    """
    Export model from PyTorch hub / TensorFlow hub and try convert the model into various format for different serving
    engines.
    """
    from modelci.hub.init_data import export_model

    export_model(model_name=name, framework=framework, enable_trt=trt)