def __init__(self, ninput, noutput, non_linear='ReLU'):
    super().__init__()
    self.conv = nn.Conv2d(ninput, (noutput - ninput), (3, 3), stride=2, padding=1, bias=False)
    self.pool = nn.MaxPool2d(2, stride=2)
    self.bn = nn.BatchNorm2d((noutput - ninput), eps=0.001)
    self.relu = NON_LINEARITY[non_linear]
def __init__(self, ninput, noutput, non_linear='ReLU'):
    super().__init__()
    self.conv = nn.Conv2d(ninput, (noutput - ninput), (3, 3), stride=2, padding=1, bias=False)
    self.pool = nn.MaxPool2d(2, stride=2)
    self.bn = nn.BatchNorm2d((noutput - ninput), eps=0.001)
    self.relu = NON_LINEARITY[non_linear]