Skip to content

反向传播与训练机制实验资产表 ​

本表登记训练机制实验的环境、工作负载、指标和证据状态。环境创建见实验 Quickstart,Profile 与隔离关系见环境资产索引,实际依赖与硬件写入结果 JSON。

直接运行环境 ​

本专题的六个机制级 GPU 实验只使用一个环境。CPU 机制练习不需要单独创建 GPU Profile;后训练和多卡实验是后续跨专题入口,不属于本专题的安装要求。

环境 ID直接入口可形成的证据
TRAIN-CPUPart 02 · 09–12、混合精度反向、55、56、19 的框架一致性核对原生算子、Accelerate、Transformers 与 checkpoint 行为
TRAIN-CUDA-MECHANISMPart 02 · 11、TRAIN-MIXED-PRECISION-BACKWARD、55、56、19、42单卡梯度累积、AMP、更新幅度、持续预训练、检查点和卸载的正确性、峰值显存与单步时间

TRAIN-CUDA-MECHANISM 不安装 Serving、多卡编排或量化产物工具。六个 GPU 实验共用 Qwen/Qwen2.5-0.5B 的同一不可变 commit,以固定文本构造训练批次;它们验证真实模型计算图上的局部机制,完整训练和策略组合进入项目页。

跨专题环境入口 ​

后续任务独立 Profile进入条件
完整 SFT 与 LoRAPT-SFT-CUDA需要真实模型、数据集和训练产物时进入后训练专题
偏好优化与 RLPT-RL-CUDA需要 rollout、奖励或策略更新时进入后训练专题
多卡同步与切分PAR-CUDA-MECHANISM需要 collective、状态切分或扩展效率时进入通信与并行专题

机制实验注册表 ​

实验 ID / 页面固定对照主要指标当前状态
TRAIN-LOCAL-BACKWARD / Part 02 · 09手写局部梯度 → Autograd输出、输入梯度、参数梯度、误差CPU 机制完整;不需要独立 GPU benchmark
TRAIN-ATTENTION-BACKWARD / Part 02 · 10手写 Attention backward → AutogradQ/K/V 梯度、保存状态、数值误差CPU 机制完整;真实 Kernel 归算子优化
TRAIN-GRADIENT-ACCUMULATION / Part 02 · 11完整批次 → 2 / 4 次微批次累积参数/loss 差异、绝对与增量峰值、step median/P95RTX 5070 Ti、Qwen2.5-0.5B 的序列 128 / 256 对照已验证;短序列为 tune,序列 256 最多降低约 17.3% 峰值并增加约 33.5% median
TRAIN-SCHEDULER-CLOCK / Part 02 · 12成功更新与跳步下的 WSD 时钟Warmup/Stable/Decay 边界、有效更新计数CPU 机制完整;无需独立 GPU benchmark
TRAIN-UPDATE-DYNAMICS / Part 02 · 55基准学习率 / 高学习率无裁剪 / 高学习率裁剪有限值、梯度范数、裁剪系数、参数增量、峰值、step median/P95GPU 实验已准备;局部稳定性不替代独立质量与保留集评测
TRAIN-CONTINUAL-PRETRAIN / Part 02 · 560 更新基线 → 固定步数全序列领域续训领域 / 保留 token loss、有限值、checkpoint 字段、峰值、step median/P95GPU 实验已准备;内置小文本只验证测量链,真实结论需版本化领域与保留集
TRAIN-AMP-BACKWARD / 混合精度反向FP32 → AMP 与溢出样本有限值、scale、跳步、峰值、单步时间RTX 5070 Ti、Qwen2.5-0.5B 的 default14 已验证;FP16 显存下降约 10.7%,时间基本持平
TRAIN-CHECKPOINT / Part 02 · 19普通保存 → 原生 gradient checkpointingloss/梯度范数、峰值、单步时间RTX 5070 Ti、Qwen2.5-0.5B 的 b2s512_default2 已验证;allocated 峰值降低约 33.7%,median 增加约 19.2%,决策为 accept
TRAIN-OFFLOAD / Part 02 · 42GPU 驻留 → host offloadloss/梯度范数、D2H/H2D、峰值、单步时间RTX 5070 Ti、Qwen2.5-0.5B 的 b2s512_default1 已验证;allocated 峰值降低约 34.3%,median 增至 5.87 倍,决策为 tune
TRAIN-STATE-LEDGER / Part 02 · 43基线生命周期 → 策略后生命周期分阶段驻留、峰值对象、理论峰值CPU 账本完整;真实峰值由 Part 02 · 19、42、76 提供

项目接口 ​

稳定项目 ID输入机制资产项目补充的证据
TRAIN-BASELINE / Part 02 · 73监督、有效更新与 AMP固定模型和数据的训练基线、吞吐与质量
TRAIN-PROFILING-DIAGNOSIS / Part 02 · 74生命周期账本与 trace 字段主导状态、关键路径和候选归因
TRAIN-MEMORY-STRATEGY / Part 02 · 76checkpoint、offload、AMP单策略与组合策略的质量—容量—时间决策

公共结果字段 ​

字段组必须记录
身份semantic_id、workload_id、run_label、页面与候选
工作负载模型 ID 与 commit、tokenizer、输入 shape、有效 batch、更新次数、dtype、随机种子
环境Profile ID、Python、PyTorch、CUDA、GPU 与显存
正确性输出/梯度误差、有限值、跳过更新与更新次数
资源与性能峰值 allocated/reserved、单步 median/P95、搬运量或重算次数
结论failure、evidence_level、decision 与下一步

Released under the MIT License.