Repository navigation
Quick Start
- Python 3.12 recommended; create an isolated environment.
- Internet access on first run to download
distilbert-base-uncasedand Python packages. - CPU is possible but slower; CUDA needs a compatible PyTorch build and driver.
py -3.12 -m venv .venv
.\.venv\Scripts\Activate.ps1
python -m pip install --upgrade pip
python -m pip install -r requirements_distilbert.txt
python DistilBERT_Colab_Fixed_2026.pyrequirements_distilbert.txt is a streamlined optional install list. The original requirements.txt additionally includes TRL/PEFT/bitsandbytes for the unfinished Phi-2 example.
Open DistilBERT_Colab_Fixed_2026.ipynb and run the cells in order. In a notebook, %pip install -r requirements_distilbert.txt requires that requirements file be uploaded into the notebook working directory.
-
support_classifier/training checkpoints andsupport_classifier/best_model/(saved classifier + tokenizer) -
support_classifier_frozen/for frozen-backbone experiment logs
These are local outputs, not committed sample models. Training results and accuracy depend on environment and training run. See Artifacts-and-Checkpoints and Troubleshooting.
Transformer Training Wiki · Implementation status: DistilBERT source available.This project is licensed under the Apache License, Version 2.0.© 2026 Sudhanshu Shukla · Licensed under the Apache License, Version 2.0
Overview
Architecture
Implemented classifier
- DistilBERT-Workflow
- Dataset-and-Labels
- Training-Configuration
- Evaluation-and-Inference
- Frozen-Backbone-Comparison
- Artifacts-and-Checkpoints
Experimental alignment draft
Operations