PyTorch 嵌入式离线语音识别示例1
PyTorch 离线语音识别示例1
加我微信hezkz17或zkhengyang可申请加入数字音频系统研究开发交流答疑群
MNIST 模型编译流程
本文基于 MNIST 模型，从模型搭建、模型训练、模型验证、推理脚本构建、ScriptModule静态模型导出、模型编译等方面，简述 PyTorch 模型导出的整体流程
1. 模型构建、训练及验证
模型构建
通过继承 nn.Module 基类的方式来构建自定义 MNIST 模型，模型结构包含卷积层、池化层、激活层以及全连接层。
import torc
立即观看
0