You can use:

raw logits (no activation function at the end, just the raw output of the last layer) + nn.CrossEntropyLoss


F_log_softmax on the model output + nn.NLLLoss


If you need to see the probabilities for debug/printing purpose:

use softmax and print the output
use exp() and print the output

Make sure to not pass the probabilities to the loss functions.