Megh_Bhalerao:

can you please tell me if the input to the nn.CrossEntropyLoss should be softmax probabilities or raw logits?


You pass in ‘raw’ logits. The criterion itself passes them through a softmax normalization.
See nn.CrossEntropyLoss 27: “This criterion combines nn.LogSoftmax() and nn.NLLLoss() in one single class.”