pytorch【模型模块】一、模型创建步骤与nn.Module
2020-02-18 本文已影响0人
米线织毛衣
一、前言
机器学习模型主要有以下几个大的模块：
前面的内容是关于数据模块，这里的内容是关于模型模块。
二、模型创建步骤
LeNet由各层组成，一个LeNet是一个nn.Module，一个卷积层是一个nn.Module，所以一个nn.Module是可以包含很多子nn.Module的。
（这里对应代码lesson-10）
三、nn.Module
所有的模型模块都是继承自nn.Module，所以需详细了解nn.Module这个类
