MicrosoftDocs / ml-basics

Exercise notebooks for Machine Learning modules on Microsoft Learn

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The torch deep learning model should not use softmax last layer + CrossEntropyLoss

yxu02 opened this issue · comments

commented

Torch CrossEntropyLoss expects logits instead of normalized probabilities.

Thanks - fixed