MiniMind 新手教程

MiniMind 新手教程

从 0 开始、用 PyTorch 原生代码训练一个约 64M 的超小语言模型。单卡 3090 大约 2 小时、约 3 块钱,就能走完预训练 + 对话微调。

PyTorch 原生 约 64M 对齐 Qwen3 2 小时 / 3 元 Apache-2.0 GitHub 仓库
56k+
GitHub Stars
64M
主线 Dense 参数
~2h
3090 复现 Zero
~3元
租卡成本估算
6400
词表大小

? 这是什么

MiniMind 不是又一个「下载现成大模型再 LoRA 微调」的项目。它把大语言模型从结构、分词、数据、预训练、指令微调,到 LoRA、蒸馏、强化学习、工具调用和部署,全部用尽量短的 PyTorch 代码摊开给你看。

作者的原话很直白:用乐高自己拼出一架飞机,远比坐在头等舱里飞行更让人兴奋。最低不到 3 块钱的服务器成本,就能亲身体验从 0 到 1 构建一个语言模型。

「2 小时」指 SFT 阶段在单张 NVIDIA 3090 上跑完 1 个 epoch 的实测耗时;「3 块钱」指对应时段的 GPU 租用成本。完整主线数据会更久、更贵。

? 建议学习路径

先跑通推理,再搞懂结构,最后才从零训练。不要一上来就开强化学习。

1 搞清它是什么→ 2 装环境→ 3 下载现成模型对话→ 4 看结构与数据→ 5 预训练 + SFT→ 6 可选:LoRA / RL / 部署
Tokenizer 6400 词表 Pretrain 学会接龙 SFT 学会当助手 可选后训练 LoRA / DPO / RL 推理与部署 CLI · WebUI · API · Ollama / vLLM 蓝框是必须走完的对话模型;绿框随时可以先体验官方权重 Tokenizer 6400 词表 Pretrain 学会接龙 SFT 学会当助手 可选后训练 LoRA / DPO / RL 推理与部署 CLI · WebUI · API 蓝框必须走完 绿框可先体验权重
没有 GPU 也能跟完本教程:第 3 章用 CPU / MPS 跑推理,训练章节当作「看懂流程」。真正从零训练建议至少一张 24GB 显存的卡。

? 你需要准备什么

  • Python 3.10+,以及能装上的 PyTorch
  • 会用终端:git、pip、进目录跑脚本
  • 推理:普通电脑即可;训练:推荐 NVIDIA GPU(作者参考环境是 3090 24GB)
  • 数据:从 ModelScope / HuggingFace 按需下载,不必一次全下