Skip to content

Use of various deep learning models to classify flowers. Models are implemented from scratch in PyTorch using only tensor operations.

Notifications You must be signed in to change notification settings

akshaykulkarni07/deep_learning_models

Repository files navigation

Implementation of various deep learning models from scratch

Models are implemented from scratch in PyTorch using only tensor operations.

Models Implemented :

  1. Logistic Regression (Not technically a deep learning model, but gives a base for Neural Networks)
  2. Deep Neural Network (Fully-connected layers only)

Note : Only binary classification done till now. Will add multi-class classification also.

Optimization Algorithms Implemented :

  1. Batch Gradient Descent
  2. Mini-batch Gradient Descent
  3. Stochastic Gradient Descent (put batch_size = 1 in mini-batch to get this)
  4. Batch Gradient Descent with Momentum
  5. Batch RMSProp
  6. Batch Adam

Training and Validation results are added in the Results folder.

About

Use of various deep learning models to classify flowers. Models are implemented from scratch in PyTorch using only tensor operations.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published