Solved by ptrblck in post #2 
              
              
                Your model already returns the log_softmax. So using another log on this tensor is not a good idea. 
You could just for the sake of debugging use softmax and then call your nll2 method to see if the model is learning. 
However, calling log on softmax might be numerically unstable and is generally no…