Tag: Vision Transformer

In this article, we train Vision Transformer on a simple malaria classification dataset, run inference, and visualize attention maps. ...

In this article, we fine tune the Vision Transformer model on a custom image classification dataset and visualize the attention maps. ...