Skip to content

Latest commit

Β 

History

2 Commits

Folders and files

NameName
Last commit message
Last commit date
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🧠 Day 03: Deep Learning & Custom PyTorch Training Loop

Python PyTorch License

A hands-on implementation of an Artificial Neural Network (ANN) using PyTorch. This project focuses on building low-level Dataset pipelines, DataLoader batching, and a custom training loop with explicit backpropagation steps.


πŸ› οΈ Key Features

  • Custom Pipeline: Built-in PyTorch Dataset and DataLoader classes for batch processing.
  • Network Architecture: Multi-layer Perceptron (MLP) incorporating BatchNorm1d, ReLU, and Dropout layers.
  • Manual Training Loop: Explicit gradient zeroing, forward pass, loss computation, backpropagation, and optimization step execution.
  • Loss Tracking: Generates and exports training vs. validation loss curve graphs.

πŸ“‚ Repository Structure

Day03_Deep_Learning_PyTorch/
β”œβ”€β”€ artifacts/
β”‚   β”œβ”€β”€ pytorch_ann_model.pth    # Model weights state dict
β”‚   └── loss_curve.png           # Loss evaluation plot
β”œβ”€β”€ nn_from_scratch.py           # PyTorch architecture & training loop script
β”œβ”€β”€ .gitignore
└── README.md


πŸš€ How to Run
1. Setup Environment
Bash
python -m venv venv
source venv/bin/activate  # On Windows: venv\Scripts\activate
pip install torch pandas numpy scikit-learn matplotlib
2. Execute PyTorch Training Loop
Bash
python nn_from_scratch.py
Part of the 7-Day Machine Learning Engineering Challenge.

About

Custom PyTorch Neural Network implementation featuring a manual training loop, regularization (Dropout & Batch Normalization), and loss curve visualization. Part of the 7-Day ML Engineering Challenge.

Topics

Resources

Stars

2 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages