MobileNetv2优化问题

 

LinearBottleneck

优化时，把卷积层2去掉时，执行速度1.2s，加上卷积层2时，执行速度是0.2s，

加卷积后；

torch.Size([1, 960, 7, 7])

去掉卷积层2后；shape

torch.Size([1, 960, 112, 112])

导致了速度变慢。

 

class LinearBottleneck(nn.Module):
    def __init__(self, inplanes, outplanes, stride=1, t=6, activation=nn.ReLU6):
        super(LinearBottleneck, self).__init__()
        self.conv1 = nn.Conv2d(inplanes, inplanes * t, kernel_size=1, bias=False)
        self.bn1 = nn.BatchNorm2d(inplanes * t)
        self.conv2 = nn.Conv2d(inplanes * t, inplanes * t, kernel_size=3, stride=stride, padding=1, bias=False,