Skip to content

Comparison of common activation functions on MNIST dataset using PyTorch.

License

Notifications You must be signed in to change notification settings

unvercan/activation-function-comparison-pytorch

Repository files navigation

activation-functions-comparison-pytorch

Comparison of common activation functions on MNIST dataset using PyTorch.

Activation functions:

  • Relu
  • Sigmoid
  • Tanh

Best result: Relu