AGI Hunt· stanfordnlp·· 3 小时前AI 评分51
斯坦福 CS336:7B 模型权重仅 14GB,训练却要 84GB 起步
斯坦福 CS336:7B 模型权重仅 14GB,训练却要 84GB 起步
AI 导读
斯坦福 NLP 官方转推 Percy Liang 的 CS336 第二讲《PyTorch 与资源核算》,配套笔记由 Chao Ma 撰写。课程算了一笔显存账,7B 参数模型的 BF16 权重约 14GB,加上 BF16 梯度和两个 FP32 Adam 动量缓冲后,还没计入激活值和临时缓冲就已需要 84GB。
来源:AGI Hunt · agihunt.info