Hi, I am new to PyTroch. I am facing an issue where when I apply softmax to predicted probabilities, all the classes are assigned the same probability. For example for a 9 class problem, the output for each class is 0.111111. This results in a constant Cross entropy loss, no matter what the input is.
Although when I take argmax of these same probabilities, the predicted classes are correct and the network has more than 90% accuracy. The individual predicted probability values are very small.