What loss we should use to train the SplitBreastModel ?

[Mahesh Bhosale]

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?

[Nan Wu]

You should use BCELoss.