Convert Model

convert_model(self, input_model_path: str, output_dir: str, target_framework: str | Framework, target_device_name: str | DeviceName, target_data_type: str | DataType = DataType.FP16, target_software_version: str | SoftwareVersion | None = None, input_layer: InputLayer | None = None, dataset_path: str | None = None, wait_until_done: bool = True, sleep_interval: int = 30) → ConverterMetadata

Convert a model to the specified framework.

• Parameters:
  • input_model_path (str) – The file path where the model is located.
  • output_dir (str) – The local folder path to save the converted model.
  • target_framework (Union *[*str , Framework ]) – The target framework name.
  • target_device_name (Union *[*str , DeviceName ]) – Target device name. Required if target_device is not specified.
  • target_data_type (Union *[*str , DataType ]) – Data type of the model. Default is DataType.FP16.
  • target_software_version (Union *[*str , SoftwareVersion ] , optional) – Target software version.
    Required if target_device_name is one of the Jetson devices.
  • input_layer (InputShape , optional) – Target input shape for conversion (e.g., dynamic batch to static batch).
  • dataset_path (str , optional) – Path to the dataset. Useful for certain conversions.
  • wait_until_done (bool) – If True, wait for the conversion result before returning the function.
    If False, request the conversion and return the function immediately.
• Raises:
  e – If an error occurs during the model conversion.
• Returns:
  Convert metadata.
• Return type:
  ConverterMetadata

Example

from netspresso import NetsPresso
from netspresso.enums import DeviceName, Framework, SoftwareVersion


netspresso = NetsPresso(email="YOUR_EMAIL", password="YOUR_PASSWORD")

converter = netspresso.converter_v2()
conversion_task = converter.convert_model(
    input_model_path="./examples/sample_models/test.onnx",
    output_dir="./outputs/converted/TENSORRT_JETSON_AGX_ORIN_JETPACK_5_0_1",
    target_framework=Framework.TENSORRT,
    target_device_name=DeviceName.JETSON_AGX_ORIN,
    target_software_version=SoftwareVersion.JETPACK_5_0_1,
)