def __init__(self, in_channels: int, out_channels: int, withBNRelu: bool=True) -> None:
    super().__init__()
    self.add_module('conv1', ConvDPUnit(in_channels, in_channels, True))
    self.add_module('conv2', ConvDPUnit(in_channels, out_channels, withBNRelu))
def __init__(self, in_channels: int, out_channels: int, withBNRelu: bool=True) -> None:
    super().__init__()
    self.add_module('conv1', ConvDPUnit(in_channels, in_channels, True))
    self.add_module('conv2', ConvDPUnit(in_channels, out_channels, withBNRelu))