ViT explanation for regression problem (no class label)?

[thaotnguyen]

I finetuned a model from ‘timm’ for a regression problem by setting ‘nb_classes’ to 1 and then using MSE loss. How can I use the methods in this repo to display a saliency map without distinct classes?