I am training a model with nn.LogSoftmax(). Well, I am actually using torch.nn.CrossEntropyLoss() as loss funtion which combines nn.LogSoftmax()  and nn.NLLLoss(). However, During the test phase, I want the model to output the probabilities, not the logs of probabilities. So, I would like to use nn.Softmax() for the test phase, not nn.LogSoftmax().
My Question: after training with nn.LogSoftmax(), can I test the model with nn.Softmax() instead of nn.LogSoftmax()? Or do I need to train the model with nn.Softmax() to be able to use nn.Softmax() during the test phase?
Thanks, Carmelo