ZHDA204A June 2026 – October 2026 MSPM0G5187
本节介绍 Tensorlab,一个面向工作流程的工具集,其对基于 PyTorch 的模型训练进行优化。通过利用核心 PyTorch 模块和类,Tensorlab 有助于神经网络的端到端设计和训练。
与高级抽象工具不同,Tensorlab 专为需要对机器学习管道进行精细端到端控制的专业开发人员而设计。Tensorlab 为高度定制的开发提供了必要的灵活性,能够精确操作数据采集、模型架构、训练超参数和编译工作流程。
以下实现将 MNIST(数字识别)数据集用作基础基准,以演示如何从第一性原理构建神经网络。技术进步从基线实现开始,并以迭代方式整合 PyTorch 生态系统的高级功能。该工作流程包含整个部署生命周期:通过 ONNX 导出模型、使用 TVM (TI NNC) 进行编译和 PC 端推理。本节最后介绍了量化感知训练 (QAT) 的实现,该训练利用 TI 特定的 NPU 量化技术来优化边缘部署的模型性能。