将 PyTorch 模型导出为 LiteRT - Qualcomm Dragonwing Documentation

将 PyTorch 模型转换为 LiteRT 的推荐方法是一个多步过程: PyTorch → ONNX → TensorFlow (SavedModel) → LiteRT

将 PyTorch 模型导出为 ONNX

使用 torch.onnx.export 将 PyTorch 模型转换为 ONNX 格式:

import torch

model.eval()
dummy_input = torch.randn(1, 3, 224, 224)
torch.onnx.export(
    model,
    dummy_input,
    "model.onnx",
    opset_version=13,
    input_names=["input"],
    output_names=["output"],
    do_constant_folding=True
)

将 ONNX 转换为 LiteRT

将模型导出为 ONNX 后,请按照将 ONNX 模型导出为 LiteRT 中的说明完成到 LiteRT 格式的转换。