Solved by ptrblck in post #11 
              
              
                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 ou…