SimMIMLinearDecoder

class mmpretrain.models.necks.SimMIMLinearDecoder(in_channels, encoder_stride)

Linear Decoder For SimMIM pretraining.

This neck reconstructs the original image from the shrunk feature map.

Parameters:

• in_channels (int) – Channel dimension of the feature map.

• encoder_stride (int) – The total stride of the encoder.

forward(x)

Forward function.