Solved by ptrblck in post #2 
              
              
                Remove the nn.Softmax at the end of your model, as nn.CrossEntropyLoss will apply F.log_softmax internally. 
PS: you can post code snippets by wrapping them into three backticks ```, which makes debugging easier