Method __init__

models/decoders/MLPDecoder.py:14–16

(self, input_dim=2048, embed_dim=768)

    """

    def __init__(self, input_dim=2048, embed_dim=768):
        super().__init__()
        self.proj = nn.Linear(input_dim, embed_dim)

    def forward(self, x):
        x = x.flatten(2).transpose(1, 2)