torchvision models were trained usingnn.CrossEntropyLoss which consists of nn.LogSoftmax and then nn.NLLLoss.
that’s why there is no softmax layer at the end.
So you only need to use nn.CrossEntropyLoss() in your model