Skip to content

Latest commit

 

History

History
10 lines (8 loc) · 248 Bytes

File metadata and controls

10 lines (8 loc) · 248 Bytes

loss-function-comparison-pytorch

Comparison of common loss functions on MNIST dataset using in PyTorch.

Loss functions:

  • L1
  • Mean Squared Error (MSE)
  • Negative Log Likelihood (NLL)
  • Cross Entropy Loss

Best result: Cross Entropy Loss