How to correctly use Cross Entropy Loss vs Softmax for classification?

后端 未结 0 1781
孤街浪徒
孤街浪徒 2020-12-22 19:01

I want to train a multi class classifier using Pytorch.

Following the official Pytorch doc shows how to use a nn.CrossEntropyLoss() after a last layer of

相关标签:
回答
  • 消灭零回复
提交回复
热议问题