Loss function of BreastPathQ fine-tuning

[bguetarni]

Hi, thank you for sharing your code and pre-trained models !

I have a question regarding the loss function used for fine-tuning the pre-trained model on the BreastPathQ dataset.
In line 387 it looks like you use the mean squarred error, although you build your model as a classifier. Is this what you wanted to do ? And if yes, could you explain why ?

Thank you.

[bguetarni]

I apologize, I realized that this dataset is effectively doing regression.