Softmax in the evaluation stage

[saeedizadi]

Hi,

I've found that you are explicitly applying log_softmax in your criterion.py file. Isn't is necessary to explicitly apply the softmax function on the output of the network in the evaluation stage? that is, before getting the maximum index for each class?

Thanks

[bodokaiser]

Hey,

yes, I think you are right. Could you test what happens if you add F.log_softmax(labels) here and make a PR out of it?