Problems with Cross Entropy Loss

2

You would have to normalize the unreduced and weighted loss with the classWeights as shown here.

1 Like