breast_cancer_classifier icon indicating copy to clipboard operation
breast_cancer_classifier copied to clipboard

Published 20 hours ago • verified publisher icon nyukat

What loss we should use to train the SplitBreastModel ?

Open bhosalems opened this issue 2 years ago • 0 comments

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?

bhosalems avatar May 11 '22 17:05 bhosalems