Solved by KFrank in post #4 
              
              
                Hi Vishak! 

No.  For reasons of numerical stability, it is better to have your model 
output the logits from its final Linear layer so that pytorch can use the 
log-sum-exp trick, either in CrossEntropyLoss or in LogSoftmax. 

Yes, looking at probabilities rather than logits could be easier to 
und…