"""downsample module."""

from torch import nn


def down_sample(in_planes: int, out_planes: int) -> nn.Module:
    """UpSample module."""
    return nn.Sequential(
        nn.Conv2d(
            in_planes, out_planes, kernel_size=4, stride=2, padding=1, bias=False
        ),
        nn.BatchNorm2d(out_planes),
        nn.LeakyReLU(0.2, inplace=True),
    )