标签 torch.onnx.export 下的文章

模型练好了,怎么用? 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

- 阅读剩余部分 -