把 PyTorch 模型变成 ONNX:导出、验证和可视化一次学会
模型练好了,怎么用?
ONNX 是什么?为什么需要它?
关键概念先扫盲
torch.onnx.export
dynamic_axes(动态轴)
onnx.checker.check_model
onnxruntime.InferenceSession
netron
代码 walkthrough:export_onnx.py
1. 加载第 1 天的模型类
2. 加载模型参数并切到评估模式
3. 导出 ONNX
4. 模型结构检查
5. 用同一张图对比 PyTorch 和 ONNX Runtime 的预测
6. 启动 netron 可视化
怎么看懂 netron 的图
界面布局:左中右三块
算子名:ONNX 的叫法和 PyTorch 不一样
连线上的形状:看懂数据怎么变
点开算子看参数
从代码到图的完整对照
踩坑记录
为什么用 imp