Just to be clear, do you mean to just add a last F.softmax layer when getting my test predictions?
And yes, I am using nn.CrossEntropyLoss, so I won’t need it to calculate the loss, but I want to be formal here and use softmax for another operation.
Thanks!