Looks like you did not preprocess your data or you’ve done it wrong, can you give more detail about your input data (shape, dtype, etc)?
By the way, there is no need to add softmax activation since torch.nn.CrossEntropyLoss() already contains it.