Repolode
raiyanyahya

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. 1

    克隆仓库

    git clone https://github.com/raiyanyahya/how-to-train-your-gpt.git
  2. 2

    创建并激活虚拟环境(Mac/Linux)

    python -m venv gpt_env && source gpt_env/bin/activate
  3. 3

    从依赖文件安装依赖

    pip install -r requirements.txt
  4. 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 数据每日更新。

相关项目