
一、前言
机器学习模型主要有以下几个大的模块:

前面的内容是关于数据模块,这里的内容是关于模型模块。
二、模型创建步骤



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


(这里对应代码lesson-10)

三、nn.Module

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



机器学习模型主要有以下几个大的模块:
前面的内容是关于数据模块,这里的内容是关于模型模块。
LeNet由各层组成,一个LeNet是一个nn.Module,一个卷积层是一个nn.Module,所以一个nn.Module是可以包含很多子nn.Module的。
所有的模型模块都是继承自nn.Module,所以需详细了解nn.Module这个类
本文标题:pytorch【模型模块】一、模型创建步骤与nn.Module
本文链接:https://www.haomeiwen.com/subject/ixcnfhtx.html
网友评论