vit-explain icon indicating copy to clipboard operation
vit-explain copied to clipboard
verified publisher icon jacobgil

ViT explanation for regression problem (no class label)?

Open thaotnguyen opened this issue 1 year ago • 0 comments

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?

thaotnguyen avatar Nov 11 '24 05:11 thaotnguyen