You use CrossEntropyLoss, which combines LogSoftmax and NLLLoss ! Therefore you can remove the nn.Softmax() from your Sequence (normally you also should give a dim e.g dim=1 in your case)!
But i don’t think this will solve your problem.