注意
截至 PyTorch 2.1,ONNX 导出器有两个版本。
torch.onnx.dynamo_export
是最新的(仍处于测试阶段)导出器,基于 PyTorch 2.0 发布的 TorchDynamo 技术。
torch.onnx.export
基于 TorchScript 后端,自 PyTorch 1.2.0 起可用。
一、torch.onnx.dynamo_export使用
在 60 分钟入门 中,我们有机会从高层次上了解 PyTorch,并训练了一个小型神经网络来对图像进行分类。在本教程中,我们将在此基础上扩展内容,介绍如何使用 TorchDynamo 和 torch.onnx.dynamo_export
ONNX 导出器将 PyTorch 中定义的模型转换为 ONNX 格式。
虽然 PyTorch 非常适合模型的开发迭代,但模型可以通过不同的格式部署到生产环境中,包括 ONNX(开放神经网络交换)!
ONNX 是一种灵活