pytorch【模型模块】一、模型创建步骤与nn.Module

一、前言

机器学习模型主要有以下几个大的模块：

前面的内容是关于数据模块，这里的内容是关于模型模块。

二、模型创建步骤

LeNet由各层组成，一个LeNet是一个nn.Module，一个卷积层是一个nn.Module，所以一个nn.Module是可以包含很多子nn.Module的。

（这里对应代码lesson-10）

三、nn.Module

所有的模型模块都是继承自nn.Module，所以需详细了解nn.Module这个类

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容