From 9ebf4896eb16dff3fe24f99fe34c33f3639864a2 Mon Sep 17 00:00:00 2001
From: Kyle1993 <344927747@qq.com>
Date: Wed, 4 Dec 2019 04:11:59 +0800
Subject: [PATCH] fix SCEModule.sSE (#113)

---
 segmentation_models_pytorch/base/modules.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/segmentation_models_pytorch/base/modules.py b/segmentation_models_pytorch/base/modules.py
index cb4d4b33..a9e9e6c0 100644
--- 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)