def __init__(self, in_channels: int, mid_channels: int, out_channels: int) -> None:
    super().__init__()
    self.add_module('conv1', nn.Conv2d(in_channels, mid_channels, 3, 2, 1, bias=True, groups=1))
    self.add_module('bn1', nn.BatchNorm2d(mid_channels))
    self.add_module('relu', nn.ReLU(inplace=True))
    self.add_module('conv2', ConvDPUnit(mid_channels, out_channels))
def __init__(self, in_channels: int, mid_channels: int, out_channels: int) -> None:
    super().__init__()
    self.add_module('conv1', nn.Conv2d(in_channels, mid_channels, 3, 2, 1, bias=True, groups=1))
    self.add_module('bn1', nn.BatchNorm2d(mid_channels))
    self.add_module('relu', nn.ReLU(inplace=True))
    self.add_module('conv2', ConvDPUnit(mid_channels, out_channels))