def __init__(self, in_channels, out_channels, kernel_size=3, padding=1):
    super().__init__()
    self.layer = nn.Sequential(nn.Conv2d(in_channels, out_channels, kernel_size, padding), nn.BatchNorm2d(out_channels))
def __init__(self, in_channels, out_channels, kernel_size=3, padding=1):
    super().__init__()
    self.layer = nn.Sequential(nn.Conv2d(in_channels, out_channels, kernel_size, padding), nn.BatchNorm2d(out_channels))