Method __init__

cldm/fet.py:37–41

(self, patch_dim, norm_layer=nn.LayerNorm)

class PatchMerging(nn.Module):
    def __init__(self, patch_dim, norm_layer=nn.LayerNorm):
        super(PatchMerging, self).__init__()
        self.patch_dim = patch_dim
        self.norm = norm_layer(4 * patch_dim)
        self.reduction = nn.Linear(4 * patch_dim, 2 * patch_dim)

    def forward(self, x):
        B, H, W, C = x.shape