pytorch学习笔记（6）：模型创建与nn.Module 一定坚持学完啊！！

YaYan233

于 2020-04-08 09:33:56 发布

阅读量860

点赞数

分类专栏： pytorch学习笔记

本文链接：https://blog.youkuaiyun.com/YaYan233/article/details/105341522

版权

本文介绍了PyTorch中的模型创建和模型容器，包括LenNet的结构，nn.Module的作用，以及nn.Sequential、nn.ModuleList和nn.ModuleDict的使用方法，如顺序包装网络层、迭代调用和索引访问。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

模型创建，模型容器

LenNet:

class LeNet(nn.model):
    def __init__(self,classes):
        super(LeNet,self).__init__()
        self.conv1=nn.Conv2d(3,6,5)
        self.conv2=nn.Conv2d(6,16,5)
        self.fc1=nn.Linear(16*5*5,120)
        self.fc2=nn.Linear(120,84)
        self.fc3=nn.Linear(84,classes)
 
    def forward(self,x):
        out=F.relu(self.conv1(x))
        out=F.max_pool2d(out,2)
        out&#

最低0.47元/天解锁文章