Should I use softmax as output when using cross entropy loss in pytorch?

This video explains
Should I use softmax as output when using cross entropy loss in pytorch?

--

Become part of the top 3% of the developers by applying to Toptal
https://topt.al/25cXVn

--

Music by Eric Matyas
https://www.soundimage.org
Track title: Puzzle Game 2

--

Chapters
00:00 Question
01:18 Accepted answer (Score 54)
01:43 Thank you

--

Full question
https://stackoverflow.com/questions/5567...

Accepted answer links:
[torch.nn.CrossEntropyLoss()]: https://pytorch.org/docs/stable/generate...
[nn.LogSoftmax()]: https://pytorch.org/docs/stable/generate...
[nn.NLLLoss()]: https://pytorch.org/docs/stable/generate...

--

Content licensed under CC BY-SA
https://meta.stackexchange.com/help/lice...

--

Tags
#python #pytorch #mnist #softmax

#avk47