美文网首页
pytorch【模型模块】一、模型创建步骤与nn.Module

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

作者: 米线织毛衣 | 来源:发表于2020-02-18 21:54 被阅读0次

一、前言

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

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

二、模型创建步骤

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

(这里对应代码lesson-10)

三、nn.Module

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

相关文章

网友评论

      本文标题:pytorch【模型模块】一、模型创建步骤与nn.Module

      本文链接:https://www.haomeiwen.com/subject/ixcnfhtx.html