Deep-Learning-Optimization-Algorithms
Deep-Learning-Optimization-Algorithms copied to clipboard
Visualization of various deep learning optimization algorithms using PyTorch automatic differentiation and optimizers.
Deep-Learning-Optimization-Algorithms
This repository contains the code for visualization of various deep learning optimization algorithms. PyTorch is used for automatic differentiation and implementing the various optimizers.
Stochastic Gradient Descent (SGD)
| 2D Convex Surface | 2D Non-Convex Surface | 3D Surface with Saddle Point |
|---|---|---|
![]() |
![]() |
![]() |
SGD with Momentum
| 2D Convex Surface | 2D Non-Convex Surface | 3D Surface with Saddle Point |
|---|---|---|
![]() |
![]() |
![]() |
Ada-delta
| 2D Convex Surface | 2D Non-Convex Surface | 3D Surface with Saddle Point |
|---|---|---|
![]() |
![]() |
![]() |
Adaptive Moment Estimation (ADAM)
| 2D Convex Surface | 2D Non-Convex Surface | 3D Surface with Saddle Point |
|---|---|---|
![]() |
![]() |
![]() |











