包含 923 份论文、课程与文章的 Markdown 机器学习知识库,并附带一个基于 NumPy 的自定义分类/回归算法 APRL。
raiyanyahya/how-to-train-your-gpt
一套带逐行注释的教程,用 Python 和 PyTorch 从零实现并训练一个类 LLaMA 架构的小型 GPT。
- Star
- 4,049
- Fork
- 505
- 未关闭 Issue
- 5
- 最近提交
- 2个月前
这是什么?
这是一个学习型项目,通过从零实现一个现代的仅解码器(decoder-only)语言模型,讲解大模型的工作原理。内容覆盖完整流程:BPE 分词器、词嵌入、旋转位置编码(RoPE)、带因果掩码的多头注意力、使用 RMSNorm 和 SwiGLU 的 Transformer 块、基于 AdamW 和余弦预热的训练循环,以及带 KV 缓存和多种采样方式的推理引擎。
教程共 12 章,按顺序阅读即可,另有约 28 篇独立的主题讲解和配套 Jupyter 笔记本。代码逐行注释,说明每一行做什么以及为什么这样写。只需要基础 Python 知识,不要求有机器学习背景。
默认配置是一个小模型(d_model=256,4 层),在 CPU 上几分钟即可训练完成。项目也提供约 1.51 亿参数的 GPT-2 规模配置,但需要 GPU。本项目面向学习用途,而非生产环境。
核心功能
- 01十二章循序渐进,从分词一直讲到推理
- 02逐行注释,说明每段代码的作用和原因
- 03实现 RoPE、RMSNorm、SwiGLU、Pre-Norm 结构和权重绑定
- 04自定义训练循环,包含 AdamW、余弦预热和混合精度
- 05推理支持 KV 缓存、temperature、top-k/top-p 采样和 beam search
- 06提供可直接运行的 main.py,默认配置可在 CPU 上训练
适用场景
- 想弄清 ChatGPT 类模型内部原理的 Python 开发者
- 从零学习 Transformer 注意力机制和训练过程的学生
- 比较 RoPE 与可学习位置编码、RMSNorm 与 LayerNorm 等架构选择的工程师
- 准备小型语言模型实验课的教师
快速上手
- 1
克隆仓库
git clone https://github.com/raiyanyahya/how-to-train-your-gpt.git - 2
创建并激活虚拟环境(Mac/Linux)
python -m venv gpt_env && source gpt_env/bin/activate - 3
从依赖文件安装依赖
pip install -r requirements.txt - 4
运行训练脚本(默认为小型配置)
python main.py
使用前须知
- 默认配置是小型模型,151M参数的GPT-2规模配置需要GPU。
- 这是学习资料,仅用于学习,不适用于生产级模型。
标签
常见问题
- how-to-train-your-gpt 是什么?
- 一套带逐行注释的教程,用 Python 和 PyTorch 从零实现并训练一个类 LLaMA 架构的小型 GPT。 这是一个学习型项目,通过从零实现一个现代的仅解码器(decoder-only)语言模型,讲解大模型的工作原理。内容覆盖完整流程:BPE 分词器、词嵌入、旋转位置编码(RoPE)、带因果掩码的多头注意力、使用 RMSNorm 和 SwiGLU 的 Transformer 块、基于 AdamW 和余弦预热的训练循环,以及带 KV 缓存和多种采样方式的推理引擎。
- how-to-train-your-gpt 可以免费使用吗?用的什么许可证?
- how-to-train-your-gpt 以 MIT 许可证开源。用于商业产品前,请先确认许可证条款。
- how-to-train-your-gpt 还在维护吗?
- 半年内有更新。how-to-train-your-gpt 最近一次提交是在 2026年8月30日。
- how-to-train-your-gpt 适合用来做什么?
- 想弄清 ChatGPT 类模型内部原理的 Python 开发者 从零学习 Transformer 注意力机制和训练过程的学生 比较 RoPE 与可学习位置编码、RMSNorm 与 LayerNorm 等架构选择的工程师 准备小型语言模型实验课的教师
- how-to-train-your-gpt 怎么安装和运行?
- 克隆仓库: git clone https://github.com/raiyanyahya/how-to-train-your-gpt.git 创建并激活虚拟环境(Mac/Linux): python -m venv gpt_env && source gpt_env/bin/activate 从依赖文件安装依赖: pip install -r requirements.txt 运行训练脚本(默认为小型配置): python main.py
- 在哪里获取 how-to-train-your-gpt?
- 源代码和安装说明在 GitHub 上:https://github.com/raiyanyahya/how-to-train-your-gpt
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
25534PythonOther更新于 2个月前
一门免费开源的 AI 工程课程,共 542 节课,用 Python、TypeScript、Rust 和 Julia 从零实现核心概念。
7.1万1.3万PythonMIT更新于 昨天
一本从零开始讲解并动手实现大语言模型的中文开源教程,涵盖 Transformer、预训练、微调、RAG 与 Agent。
3.4万3253Jupyter NotebookOther更新于 2个月前
54 个无依赖、单文件的 Python 脚本,从零实现现代 AI 核心算法,便于读懂其底层原理。
1433105PythonMIT更新于 5天前
面向文本、视觉、音频和多模态的前沿机器学习模型定义框架,可用于推理和训练。
16.8万3.5万PythonApache-2.0更新于 10小时前