CrossEntropyLoss expects scores for each class as its input.
You are using softmax in your model, so just remove it.
Alternatively you could change it to F.log_softmax and use NLLLoss.
Let me know, if your model is learning!