def __init__(self, c1, c2) -> None:
    super().__init__()
    self.conv_3x3 = ConvModule(c1, c2, 3, 1, 1)
    self.attention = nn.Sequential(nn.AdaptiveAvgPool2d(1), nn.Conv2d(c2, c2, 1, bias=False), nn.BatchNorm2d(c2), nn.Sigmoid())
def __init__(self, c1, c2) -> None:
    super().__init__()
    self.conv_3x3 = ConvModule(c1, c2, 3, 1, 1)
    self.attention = nn.Sequential(nn.AdaptiveAvgPool2d(1), nn.Conv2d(c2, c2, 1, bias=False), nn.BatchNorm2d(c2), nn.Sigmoid())