explaingit

rasbt/deeplearning-models

17,501Jupyter NotebookAudience · researcherComplexity · 2/5Setup · moderate

TLDR

A collection of Jupyter Notebooks implementing classic deep learning architectures, from basic perceptrons to CNNs like AlexNet and MobileNet, in both TensorFlow and PyTorch, with code and explanations side by side.

Mindmap

mindmap
  root((deeplearning-models))
    What it is
      Notebook collection
      Reference implementations
    Topics
      Perceptron basics
      Multilayer networks
      CNNs
    Architectures
      AlexNet
      DenseNet
      MobileNet
      LeNet-5
    Frameworks
      PyTorch
      TensorFlow
      PyTorch Lightning
    Datasets
      MNIST
      CIFAR-10
      Iris
Click or tap to explore — scroll the page freely

Code map

Detail Auto

An interactive map of this repo's files and how they connect — its source is parsed live in your browser. Click Visualize to build it.

filefunction / class

Things people build with this

USE CASE 1

Study how a specific neural network architecture such as LeNet-5 or DenseNet works by running a step-by-step notebook instead of reading the paper alone.

USE CASE 2

Compare the same model implemented in TensorFlow versus PyTorch side by side to decide which framework to adopt.

USE CASE 3

Reuse a reference implementation of a named architecture as a starting point for a new deep learning project or course assignment.

Tech stack

PythonPyTorchTensorFlowPyTorch LightningJupyter Notebook

Getting it running

Difficulty · moderate Time to first run · 30min

Requires Python with PyTorch or TensorFlow installed, GPU speeds up training but is not required for smaller notebooks.

In plain English

This repository is a collection of deep learning architectures, models, and tips packaged as Jupyter Notebooks. Jupyter Notebooks are interactive documents that mix code, charts, and explanations, which makes them well suited for learning how a model works step by step. The README presents implementations across two of the main deep learning frameworks: TensorFlow and PyTorch (with some additional examples using PyTorch Lightning), so the same idea is often shown in more than one style. The notebooks are organised by topic. There is a section on traditional machine learning starters such as the Perceptron, Logistic Regression, and Softmax Regression, trained on simple datasets like 2D toy data, the Iris dataset, and MNIST (a classic dataset of handwritten digits). A multilayer perceptron section adds variations like dropout, batch normalization, and a from-scratch backpropagation implementation. The largest area covers convolutional neural networks, which are models specialised for image data: basic CNNs, weight-initialisation variants, and named architectures including AlexNet, DenseNet-121, an All-Convolutional Net, LeNet-5, MobileNet v2 and v3, and Network in Network, with examples trained on datasets such as MNIST, CIFAR-10, and QuickDraw. Someone would use this when they want to study or reuse reference implementations of well-known deep learning models side by side, rather than reading papers or building from scratch. The tech stack stated in the README is Python notebooks running on TensorFlow, PyTorch, and PyTorch Lightning. The full README is longer than what was provided.

Copy-paste prompts

Prompt 1
I want to understand how a Convolutional Neural Network classifies images. Walk me through the CNN notebook in deeplearning-models step by step, explaining each layer's job in plain English.
Prompt 2
Show me how to adapt the MobileNet v2 notebook from deeplearning-models to classify my own image dataset instead of CIFAR-10, including how to change the data loader.
Prompt 3
I am learning backpropagation. Explain the from-scratch backprop notebook in deeplearning-models using simple analogies, then show me exactly where the weight update happens in the code.
Prompt 4
Compare the AlexNet implementation in this repo with the original 2012 paper. What architecture simplifications were made and why?
Open on GitHub → Explain another repo

← rasbt on gitmyhub — every repo by this author, as a profile.

Verify against the repo before relying on details.