#202 · Primary category: Deep Learning Frameworks
knowledge-distillation-pytorch
A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility
Project last updated:03/25/23
GitHub Stars
2.0K
Forks
353
Contributors
5
License
MIT
Why we included this project
This repo is a clean, self-contained way to learn knowledge distillation by actually running it. It trains a small student model on the logits of a larger teacher, with two worked examples on CIFAR-10: a five-layer CNN distilled from a ResNet-18, and a ResNet-18 distilled from a ResNeXt-29. Each experiment reads its settings from a single params.json file, so adjusting the temperature or distillation weight is a quick edit instead of a long command line. The repo also ships scripts for hyperparameter search and result synthesis, which turn the usual fiddly tuning into something repeatable, plus pretrained teacher checkpoints you can download. Anyone studying model compression or wanting to reproduce classic KD results can get concrete numbers without rebuilding the whole setup from scratch.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.