e0b84042ebe4e182b6dd7ab1673e00e0fbc3c8a6,pytorch/pytorchcv/models/common.py,SEBlock,__init__,#SEBlock#Any#Any#Any#Any#Any#,946

Before Change

            in_channels=mid_channels,
            out_channels=channels,
            bias=True)
        self.sigmoid = HSigmoid() if approx_sigmoid else nn.Sigmoid()

    def forward(self, x):
        w = self.pool(x)
        w = self.conv1(w)

After Change

            out_channels=channels,
            bias=True)
        // self.sigmoid = HSigmoid() if approx_sigmoid else nn.Sigmoid()
        self.sigmoid = get_activation_layer(out_activation)

    def forward(self, x):
        w = self.pool(x)
        w = self.conv1(w)

Instances