lukysummer / VisionTransformer-PyTorch

ViT fine-tuning and iterative erasing prediction algorithm using self-attention weights in PyTorch

Vision-Transformer-in-PyTorch

ViT fine-tuning and iterative erasing prediction algorithm using self-attention weights (in PyTorch)

You can see :

• My Medium Post for fine-tuning ViT here
• My Medium Post for iterative erasing prediction algorithm for ViT here