I have the same problem since two days ago and I do not find the cause of the problem. I even deleted the logsoftmax from my code and still the same…
Please tell me if you find a solution and I will do the same if I do

image1013×104 42.1 KB