Hi,
I would like to use the ViT model for classification and adapt it to a regression task, is it feasible ?
Can the model work just by changing the loss function ? How can I define the classes in the _info method of my custom dataset since there is an infinity of them possible ? What are all the other changes to make ?
Thank you