search

srinidhiPY / SSL_CR_Histo

Official code for "Self-Supervised driven Consistency Training for Annotation Efficient Histopathology Image Analysis" Published in Medical Image Analysis (MedIA) Journal, Oct, 2021.
https://doi.org/10.1016/j.media.2021.102256
MIT License
62 stars 21 forks source link

Loss function of BreastPathQ fine-tuning #9

Closed bguetarni closed 1 year ago

bguetarni commented 1 year ago

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 commented 1 year ago

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