No, you used softmax as a layer inside your network and wrong loss for that;
You should not apply softmax with crossentropyloss, but when you are just doing forward and not training you can apply softmax (argmax of something is equal to the argmax of the softmax of that same thing)