search

nyukat / breast_cancer_classifier

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening
https://ieeexplore.ieee.org/document/8861376
GNU Affero General Public License v3.0
844 stars 269 forks source link

What loss we should use to train the SplitBreastModel ? #57

Closed bhosalems closed 2 years ago

bhosalems commented 2 years ago

I see that log_softmax is calculated in forward() of output layer used by all the models. Should we be using BCELoss or CrossEntopyloss in that case?

wooginawunan commented 2 years ago

You should use BCELoss.