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

一、前言

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

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

二、模型创建步骤

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

（这里对应代码lesson-10）

三、nn.Module

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