r0l1
diff --git a/‎segmentation_models_pytorch/base/modules.py
Copy file name to clipboardExpand all lines: segmentation_models_pytorch/base/modules.py
+1-1Lines changed: 1 addition & 1 deletion b/‎segmentation_models_pytorch/base/modules.py
Copy file name to clipboardExpand all lines: segmentation_models_pytorch/base/modules.py
+1-1Lines changed: 1 addition & 1 deletion
@@ -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)`