LogoCookLLM文档
LogoCookLLM文档
首页CookLLM

原理精讲

词元化
Tokenization 基础BPE 算法详解GPT 系列 TokenizerBPE 训练工程化
模型架构
Attention 机制详解
位置编码
位置编码基础RoPE 数学推导RoPE 代码实现长度外推
GPU 编程基础
GPU 架构基础张量布局Triton 入门:向量加法
FlashAttention
Flash Attention 原理详解从朴素实现到 Auto-TuningBlock Pointer 与多维支持Causal Masking 优化Grouped Query Attention反向传播实现
分布式训练数据并行ZeRO 优化器全分片数据并行张量并行流水线并行多维混合并行

动手训练

概述
预训练
预训练数据Tokenizer 训练模型架构数据流水线训练循环监控与验证
X (Twitter)
预训练

训练循环

会员专享

拆开 LightningCLI、PretrainModule、优化器和调度器

登录以继续阅读

这是一篇付费内容,请登录您的账户以访问完整内容。

数据流水线

理解 Parquet shard 如何变成 input_ids、labels 和 attention_mask

监控与验证

使用 TensorBoard、SwanLab、采样文本和 checkpoint 跟踪预训练

目录

训练入口
LightningCLI 是什么
Lightning 接管了什么
启动脚本
配置合并
Training step
Validation step
优化器
WSD 学习率调度
Trainer 配置
常用覆盖参数