Skip to content

Comparison of common loss functions in PyTorch using MNIST dataset

License

Notifications You must be signed in to change notification settings

unvercan/loss-function-comparison-pytorch

Repository files navigation

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