nn.CrossEntropyLoss for binary classification didn’t work for me too! In fact, it did the opposite of learning. Why didn’t it work for you? Can you please explain the behavior I am observing?
Note: The same model with nn.MSELoss worked fine (passed overfitting test). And I am supplying the unnormalized FC layer output to the CE loss function.
image1450×702 57.7 KB