output not feed into the softmax

[LilioISnotAfox]

Hello, I have noticed that in main_tasn.py, the outputs of fc directly are used to compute loss without feeding into the softmax, Is there any reason or just a mistake?

[Ahzyuan]

I think the 'nn.CrossEntropyLoss()' in the later procedure will apply softmax to the output

[xxx0090]

I also found it. Do you solve it?