fix SCEModule.sSE (#113)

Kyle1993 · qubvel · commit 9ebf4896eb16 · 2019-12-03T23:11:59.000+03:00
diff --git a/segmentation_models_pytorch/base/modules.py b/segmentation_models_pytorch/base/modules.py
@@ -58,7 +58,7 @@ def __init__(self, in_channels, reduction=16):
             nn.Conv2d(in_channels // reduction, in_channels, 1),
             nn.Sigmoid(),
         )
-        self.sSE = nn.Sequential(nn.Conv2d(in_channels, in_channels, 1), nn.Sigmoid())
+        self.sSE = nn.Sequential(nn.Conv2d(in_channels, 1, 1), nn.Sigmoid())
 
     def forward(self, x):
         return x * self.cSE(x) + x * self.sSE(x)

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ def __init__(self, in_channels, reduction=16):`
`58`	`58`	`nn.Conv2d(in_channels // reduction, in_channels, 1),`
`59`	`59`	`nn.Sigmoid(),`
`60`	`60`	`)`
`61`		`- self.sSE = nn.Sequential(nn.Conv2d(in_channels, in_channels, 1), nn.Sigmoid())`
	`61`	`+ self.sSE = nn.Sequential(nn.Conv2d(in_channels, 1, 1), nn.Sigmoid())`
`62`	`62`
`63`	`63`	`def forward(self, x):`
`64`	`64`	`return x * self.cSE(x) + x * self.sSE(x)`