Pytorch中nn.Sequential()函数创建网络的几种方法
创始人
2025-01-11 08:36:00
0

1. 创作灵感

在创建大型网络的时候,如果使用nn.Sequential()将几个有紧密联系的运算组成一个序列,可以使网络的结构更加清晰。

2.应用举例

为了记录nn.Sequential()的用法,搭建以下测试网络:

2.1 方法一

把网络分成3个Sequential序列,分别实现:第一个是卷积序列,第二个是铺平成一维的操作,第3个包含了两个线性层。

 class TestNet(nn.Module):     def __init__(self, in_channels, mid_channels):         super(TestNet,self).__init__()         self.conv = nn.Sequential(             nn.Conv2d(in_channels,in_channels*2,kernel_size=3,stride=2,padding=1),             nn.BatchNorm2d(in_channels*2),             nn.ReLU(inplace=True),             nn.Conv2d(in_channels*2,in_channels*4,kernel_size=3,stride=2,padding=1),             nn.BatchNorm2d(in_channels*4),             nn.ReLU(inplace=True)         )         self.flat = nn.Flatten()         self.linaer = nn.Sequential(             nn.Linear(49*4, 64),             nn.Linear(64, 10),             nn.Linear(10, 1)         )      def forward(self,x):         x = self.conv(x)         print(x.shape)         x = self.flat(x)         print(x.shape)         x = self.linaer(x)         print(x.shape)         return x 

运行结果:

从维度上判断,网络符合预期。

2.2  方法二

第二种方法,在网络结构比较复杂且重复的单元比较多,为了自动化生成网络,通常会先定义一个列表,在列表中添加网络,再使用nn.Sequential()。

使用方法二所需要的代码如下:

class TestNet2(nn.Module):     def __init__(self, in_channels):         super(TestNet2,self).__init__()         layer1 = []         layer2 = []         layer1.append(nn.Conv2d(in_channels,in_channels*2,kernel_size=3,stride=2,padding=1))         layer1.append(nn.BatchNorm2d(in_channels*2))         layer1.append(nn.ReLU(inplace=True))         layer1.append(nn.Conv2d(in_channels*2,in_channels*4,kernel_size=3,stride=2,padding=1))         layer1.append(nn.BatchNorm2d(in_channels*4))         layer1.append(nn.ReLU(inplace=True))         self.conv = nn.Sequential(*layer1)         self.flat = nn.Flatten()         layer2.append(nn.Linear(49*4, 64))         layer2.append(nn.Linear(64, 10))         layer2.append(nn.Linear(10, 1))         self.linaer = nn.Sequential(*layer2)      def forward(self,x):         x = self.conv(x)         print(x.shape)         x = self.flat(x)         print(x.shape)         x = self.linaer(x)         print(x.shape)         return x

运行结果如下:

与第一种方法的结果相同。

参考文献:

nn.Sequential、nn.ModuleList、nn.ModuleDict区别及使用技巧-CSDN博客

相关内容

热门资讯

目前!聚友棋牌辅助透视挂,掌乐... 目前!聚友棋牌辅助透视挂,掌乐麻将确实有挂,真是有挂方针一、聚友棋牌游戏安装教程牌型概率发牌机制”必...
相较于以往!彩虹辅助透视挂,同... 相较于以往!彩虹辅助透视挂,同仁棋牌是有挂,一贯确实有挂同仁棋牌是不是有人用挂微扑克wpk插件教程:...
现就发布提示!腾跃辅助透视挂,... 现就发布提示!腾跃辅助透视挂,雀跃存在有挂,其实有挂工具1、进入游戏-大厅左侧-新手福利-激活码辅助...
据统计!金牛大厅辅助透视挂,哥... 据统计!金牛大厅辅助透视挂,哥哥打大a真的有挂,其实证实有挂1、金牛大厅辅助器安装包、金牛大厅辅助器...
插件辅助挂!钢铁侠辅助透视挂,... 插件辅助挂!钢铁侠辅助透视挂,飞飞黄梅棋牌是有挂,果然有挂细节黄梅棋牌脚本下载中分为三种模型:黄梅棋...
更值得关注的是!决胜奕福辅助透... 更值得关注的是!决胜奕福辅助透视挂,家家乐牌吧确实有挂,一贯详细教程1、首先打开家家乐牌吧辅助器下载...
2026版教学!518互娱辅助... 2026版教学!518互娱辅助透视挂,小南四川长牌是真的有挂,竟然有挂秘诀1、进入到小南四川长牌是否...
网友热议!掌上风云辅助透视挂,... 网友热议!掌上风云辅助透视挂,指尖四川麻将真的有挂,一直有挂方略1、四川麻将辅助软件下载优化,四川麻...
近年来!唯思竞技麻将辅助透视挂... 近年来!唯思竞技麻将辅助透视挂,正宗钟祥麻将真的是有挂,竟然有挂攻略1、唯思竞技麻将模拟器是什么优化...
截至发稿!涡阳茶馆辅助透视挂,... 截至发稿!涡阳茶馆辅助透视挂,小南四川长牌是真的有挂,一直的确有挂运小南四川长牌辅助工具,进入游戏界...