Do I need to use softmax before nn.CrossEntropyLoss()?

2

You just send the raw output. Check out this this thread for small discussion about this topic: VGG output layer - no softmax?

3 Likes