可以使用 torch.onnx.export() 函数将 torch.jit.trace() 后的模型转换为 ONNX 格式。示例代码如下:
import torch
import torch.onnx
traced_model = torch.jit.trace(model, input)
torch.onnx.export(traced_model, input, "model.onnx")
其中,model 是需要转换的 PyTorch 模型,input 是一个输入样本,"model.onnx" 是输出文件的文件名。