Multiclass classification with softmax

[KFrank]

Hi jpj!

@jpj

Or is there any better method?

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

Best.

K. Frank