/loss-function-comparison-pytorch

Comparison of common loss functions in PyTorch using MNIST dataset

Primary LanguagePythonMIT LicenseMIT

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