paulgavrikov/in1k_head_for_vit_base_patch16_224.mae

This is a linear classification head for ImageNet-1k for use with timm's vit_base_patch16_224.mae (Masked Auto Encoders). It does not contain any additional normalization or pooling layers. I trained it for 100 epochs on ImageNet-1k but still got a relatively poor performance (52.19).