Thanks a lot. Indeed I made a mistake there, I’ve already changed it as you can see in my code, I use “None” instead of “ReLU” after the last fully connected layer in PyTorch, because I’m using CrossEntropyLoss loss function so I assume I don’t need “Softmax” activation here, right?