混合精度训练(Mixed-Precision Training)是结合使用16位和32位浮点数加速模型训练、减少显存占用的方法,其核心构成包括半精度计算、损失缩放与全精度权重保留,解决的是训练效率低、显存占用过大导致模型训不动的问题,并不解决算子数值不稳定、收敛精度受损的问题。
混合精度训练是什么?
混合精度训练(Mixed-Precision Training)是结合使用16位和32位浮点数加速模型训练、减少显存占用的方法,其核心构成包括半精度计算、损失缩放与全精度权重保留,解决的是训练效率低、显存占用过大导致模型训不动的问题,并不解决算子数值不稳定、收敛精度受损的问题。
🔗 本文链接