MiniMind
从零手写的大模型训练项目
MiniMind用PyTorch原生实现从Tokenizer训练到预训练、SFT、LoRA、DPO、蒸馏的完整链路,最小模型26M参数,普通显卡就能跑。代码不依赖第三方高层接口,是学习大模型原理和快速做实验的极简方案。

核心功能
极低成本从零训练
单卡3090上跑完一个epoch约2小时,GPU租用成本3块钱左右,个人开发者也能负担。
完整训练链路开源
覆盖预训练、SFT监督微调、LoRA微调、DPO偏好对齐和模型蒸馏全过程代码。
原生PyTorch实现
所有核心算法从零手写,不依赖transformers/trl/peft等高层抽象接口,代码逐行可读。
多模态扩展支持
提供MiniMind-V视觉语言模型版本,65M参数即可完成识图和对话任务。
对齐Qwen3生态
主线结构对齐Qwen3/Qwen3-MoE,支持Dense和MoE两种模式,方便迁移和对比学习。
内置MoE混合专家
在前馈网络中引入混合专家模块,用更少计算资源提升小模型的学习效率。
产品特色
- 01
代码是真的“从零写”,不是调库封装
很多号称从零实现的项目,打开一看还是靠transformers和peft在跑。MiniMind不一样,注意力机制、RMSNorm、RoPE位置编码、MoE路由这些核心模块都是手写的,整个训练脚本加起来能在一屏内看完。想搞懂每一行代码在干什么,这个项目是很合适的起点。
- 02
门槛低到有点意外,普通显卡就能跑
最小模型26M参数,推理只占0.5GB显存,训练在单张3090上2小时跑完。整个流程不需要多卡集群,不需要几百G显存,一台普通的带独显的电脑就能上手。对想动手但设备有限的人来说,这一点挺重要的。
- 03
不光教“怎么做”,还教“为什么这么做”
项目里能看到很多设计取舍的讨论,比如为什么参数量固定时深而窄的结构往往比宽而浅的好,d_model小于512时词嵌入维度会成为瓶颈。这些不是教科书上的结论,而是在实际训练中观察到的现象,读起来比纯教程有收获。
- 04
小模型的缺点也没藏着掖着
官方文档里坦率地讨论了小模型的能力边界,知识问答容易出现幻觉,代码题可能出错,生成后段有时会重复退化。这种诚实的呈现方式反而让人觉得可信,不会期望过高然后失望。拿它当一个学习和实验的平台是合理的定位。
应用场景
✅ 大模型入门学习
配合教程从Tokenizer一路走到RLHF,把每个模块的源码逐行读懂,适合零基础起步。
✅ 高校教学实验
学生可以在个人电脑上完成从数据处理到模型对齐的完整实验,不用排队等集群资源。
✅ 算法快速验证
想测试一个新的注意力变体或训练策略,先在26M的小模型上跑通再迁移到大模型。
✅ 面试项目储备
亲手训练过一个小语言模型并理解全流程,面试聊技术细节时比只调过API有说服力。
✅ 边缘设备推理实验
26M模型推理占用极低,适合在资源受限的环境下验证轻量化模型的部署效果。
✅ 技术博客素材
把训练过程和数据记录下来,本身就是一篇有实操细节的技术分享内容。
常见问题
解答您关于"MiniMind"的常见疑问,让您更好地使用这款AI工具
-
大多数项目是拿现成的大模型做LoRA微调,本质还是在用别人的模型。MiniMind是从零开始构建模型结构、训练Tokenizer、跑预训练和微调的完整链路,代码里没有黑盒。如果你想理解大模型到底是怎么被训练出来的,而不是只会调接口,两者的差别挺大的。
-
这个成本指的是GPU租用费。在单张NVIDIA 3090上跑完预训练加SFT一个epoch,大约2小时出头,按3090租卡单价1.3元/小时算下来不到3块钱。前提是你已经有代码和数据集,租个GPU实例直接跑就行。如果用自己的显卡,那连这3块钱都省了。
-
说实话不能指望它像ChatGPT那样回答问题。它的定位是让你亲手跑通训练流程,看到一个小模型从随机参数变成能对话的模型,理解其中每一步发生了什么。知识问答、代码生成这些任务它做得不太好,但用来学习原理、做教学演示、验证训练策略的可行性,是完全够用的。
-
需要一些Python基础,至少能看懂代码在做什么。不过项目本身代码量不大,结构也清晰,配合社区里的学习教程(比如learn-minimind那个仓库有22节课),边看边跑是有可能跟下来的。如果完全没接触过深度学习,建议先补一下Transformer和PyTorch的基础。
-
支持。项目的数据集以中文为主,Tokenizer也支持中文分词。训练出来的模型用中文对话没问题,知识问答方面对中文常识的覆盖比英文要好一些。官方也提供了在线的Demo可以直接体验效果。
-
项目基于Apache 2.0协议开源,可以商用。但需要注意,你训练出来的模型能力有限,是否适合实际业务场景需要自己评估。拿来学习、研究、教学或者做技术验证都没问题。
立即体验“MiniMind”
MiniMind是一个开源的大语言模型训练项目,完全从零用PyTorch实现,最小版本只有26M参数,3块钱GPU租用成本加上2小时训练时间就能跑通预训练到微调的全流程。代码没有依赖高层抽象封装,每一步都看得见,适合想真正搞懂大模型底层原理的开发者、学生和转行者。

