The torch deep learning model should not use softmax last layer + CrossEntropyLoss
yxu02 opened this issue · comments
Torch CrossEntropyLoss expects logits instead of normalized probabilities.
Thanks - fixed
Exercise notebooks for Machine Learning modules on Microsoft Learn
yxu02 opened this issue · comments
Torch CrossEntropyLoss expects logits instead of normalized probabilities.
Thanks - fixed