Relevance score in ViT model

[Kallel_Mariem]

i work on a pretrained vit model for image classification and i want to store the relevance scores in the forward pass how can i do that ?