混合精度训练是怎么做的?

一句话速记:前向反向用 bf16/fp16 算,fp32 保存主权重和优化器状态,fp16 还需动态 loss scaling。

混合精度

同章节的其他八股