您的位置:首页 > 游戏 > 游戏 > 昇思MindSpore学习入门-模型模块自定义

昇思MindSpore学习入门-模型模块自定义

2024/9/24 23:27:22 来源:https://blog.csdn.net/buw369521/article/details/140258393  浏览:    关键词:昇思MindSpore学习入门-模型模块自定义

基础用法示例

神经网络模型由各种层(Layer)构成,MindSpore提供构造神经网络层的基础单元Cell,基于Cell进行神经网络封装。下面使用Cell构造经典模型AlexNet。

如图所示,AlexNet由5个卷积层与3个全连接层串联构成,我们使用mindspore.nn提供的神经网络层接口进行构造。

下面的代码展示了如何使用nn.Cell快速构造AlexNet。其中:

  • 顶层神经网络继承nn.Cell,为嵌套结构;
  • 每个神经网络层都是nn.Cell的子类;
  • nn.SequentialCell可以在定义顺序结构的模型时进行简化。

在定义模型的过程中,construct方法内可使用Python语法进行模型结构的任意构造,如条件、循环等控制流语句。但在进行即时编译(Just In Time)时,需通过编译器进行语法解析,此时存在语法限制,具体参考:静态图语法支持。

完成模型构造后,我们构造一个单样本数据,将其送入实例化的AlexNet中,即可求得正向结果。

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com