tom:

softmax and 1 component


Hi, thank you for your reply!
I am not sure I fully understand your answer.
When I do nn.Softmax() then I get vectors including 2 different probabilities since it is binary segmentation model.
Softmax function solely does not pick the highest probability index to compare with my target values, so surely I have to use functions like argmax after softmax function?