Resnet网络残差连接与relu顺序

最新推荐文章于 2024-07-17 20:41:09 发布

xiaohanabc

最新推荐文章于 2024-07-17 20:41:09 发布

阅读量3.2k

点赞数 3

分类专栏：深度学习

本文链接：https://blog.csdn.net/u012189988/article/details/99711324

版权

在Resnet的Bottleneck模块中，必须先进行残差连接，然后应用ReLU激活函数。若顺序调换，则会导致模型无法收敛，准确率停留在50%-70%区间。

摘要由CSDN通过智能技术生成

class Bottleneck(nn.Module):
expansion = 4

def __init__(self, inplanes, planes, stride=1, downsample=None):
    super(Bottleneck, self).__init__()
    self.conv1 = nn.Conv2d(inplanes, planes, kernel_size=1, bias=False)
    self.bn1 = nn.BatchNorm2d(planes)
    self.conv2 = nn.Conv2d(planes, planes, kernel_size=3, stride=stride,
                           padding=1, bias=False)
    self.bn2 = nn.BatchNorm2d(planes)
    self.conv3 = nn.Conv2d(planes, planes * 4, kernel_size=1, bias=False)
    self.bn3 = nn.BatchNorm2d(planes * 4)
    self.relu = nn.ReLU(inplace=True)
    self.downsample = downsample
    self.stride = stride

def forward(self, x):
residual = x

    out = self.conv1(x)
    out = self.bn1(out)
    out = self.relu(out)

    out = self.conv2(out)
    out = self.bn2(out)
    out = self.relu(out)

    out