OliveModels

The following models are available in Olive.

ONNX Model

class olive.model.ONNXModel(model_path: str | None = None, name: str | None = None, version: int | None = None, is_file: bool = True, is_aml_model: bool = False, inference_settings: dict | None = None)

OpenVINO Model

class olive.model.OpenVINOModel(model_path: str, name: str | None = None, is_file=False, version: int | None = None, is_aml_model: bool = False)

PyTorch Model

class olive.model.PyTorchModel(model_path: str | None = None, name: str | None = None, version: int | None = None, is_file: bool = False, is_aml_model: bool = False, model_loader=None, model_script=None, script_dir=None)

SNPE Model

class olive.model.SNPEModel(input_names: List[str], input_shapes: List[List[int]], output_names: List[str], output_shapes: List[List[int]], model_path: str | None = None, is_aml_model: bool = False, name: str | None = None, version: int | None = None)