Skip to content

Latest commit

 

History

History
28 lines (18 loc) · 2.38 KB

File metadata and controls

28 lines (18 loc) · 2.38 KB

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