课程学习目标:构建一个用于Mnist数据集分类的神经网络模型。

关于网络构建

网络构建是指在计算机网络中,通过编程、配置硬件设备或其他方式生成、修改和管理网络的过程。

神经网络模型是由神经网络层和Tensor操作构成的。在MindSpore框架中,神经网络模型的构建是以Cell类为基础的。一个神经网络模型表示为一个Cell,它由不同的子Cell构成。一个复杂的神经网络模型可以被看作是由多个Cell对象(子Cell)组成的嵌套结构。每个Cell可以包含其他Cell作为其子组件,从而允许构建任意复杂的网络架构。使用这样的嵌套结构,可以简单地使用面向对象编程的思维,对神经网络结构进行构建和管理。

实战

定义一个继承自nn.Cell的类,并在其中构建网络结构。

使用nn.SequentialCell来替代手动定义多个层并串联它们。

添加Softmax

 

使用model.parameters_and_names()来获取模型中的所有参数及其名称。

 

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐