Hi jpj!



 jpj:

Or is there any better method?


Yes, you should use CrossEntropyLoss 2 as your loss criterions (and not
use Softmax because CrossEntropyLoss expects logits rather than
probabilities as its predicted input).
Best.
K. Frank