I know nn.CrossEntropyLoss() incoporates softmax. So my model output did not add softmax layer. But when I do the loss.backward(), should I change the outputs using softmax like outputs = torch.softmax(outputs, dim=1).cpu().detach().numpy()?