Solved by LeviViana in post #2 
              
              
                This tutorial uses torch.nn.CrossEntropyLoss() as the loss function, which means that you should just run torch.nn.functional.softmax on your outputs, since it learned log-softmax values.