paulgavrikov

in1k_head_for_vit_base_patch16_224.mae

paulgavrikov/in1k_head_for_vit_base_patch16_224.mae

in1k_head_for_vit_base_patch16_224.mae is a machine learning model from paulgavrikov. Use it for the machine learning task on the model card, and read the license before you ship it in a product.

This is a linear classification head for ImageNet-1k for use with timm's vitbasepatch16224.mae (Masked Auto Encoders). It does not contain any additional normalization or pooling layers. I trained it for 100 epochs on…