nanoGPT

已跟踪

一个简洁且高效的仓库,用于训练与微调中等规模的 GPT 模型,适合教学与实验。

作者 Andrej Karpathy 开源时间 2022-12-28 最近提交 未知

nanoGPT 由 Andrej Karpathy 创建,是用于训练和微调中等规模 GPT 模型最简洁、最快速的代码仓库。凭借紧凑、高度可读的代码库和极少的依赖,它使 Transformer 训练流程、数据预处理和优化技术对学习者和从业者都触手可及。

极简训练流水线

  • 将 GPT 训练流水线精简到最核心的部分,每一行代码都可理解且不失正确性
  • 开箱即用地支持从头训练和在较小数据集上微调,便于快速实验
  • 示例配置和训练脚本确保可复现性,任何人都可以复现已发布的训练流程和结果

教学与原型开发

  • 广泛用作教学工具,帮助深入、实践地理解 GPT 架构和完整训练流程
  • 研究人员依赖它在受控环境中快速原型化训练实验和基准测试优化技术
  • 小团队可在不受大型训练框架开销困扰的情况下探索模型能力和数据处理策略

清晰可读的代码库

  • Python 代码库优先考虑可读性和可实验性,从初学者到中级开发者都易于上手
  • 项目采用 MIT 许可证,社区活跃,持续贡献改进和扩展
  • AI 教育和小规模模型探索中被引用最广泛的仓库之一