Method __init__

method/nn_util.py:41–45

(self, embed_dim, embed_dim_next)

class ChannelEmbeddingCompression(nn.Module):
    def __init__(self, embed_dim, embed_dim_next):
        super().__init__()
        self.patch_unembed = PatchUnEmbedIR(embed_dim=embed_dim)
        self.conv = nn.Conv2d(embed_dim, embed_dim_next, 1, 1, 0)
        self.patch_embed = PatchEmbedIR(embed_dim=embed_dim_next)

    def forward(self, x, x_size):
        x = self.patch_unembed(x, x_size)