Method __init__

modules/VIPTRv2.py:556–561

(self, dim, out_dim, norm_layer=nn.LayerNorm)

    """

    def __init__(self, dim, out_dim, norm_layer=nn.LayerNorm):
        super().__init__()
        self.dim = dim
        self.reduction = nn.Conv2d(dim, out_dim, 3, (2, 1), 1)
        # self.norm = nn.BatchNorm2d(out_dim)
        self.norm = nn.LayerNorm(out_dim)

    def forward(self, x):
        '''