pytorch_dynamic_loss_weighting

指导在PyTorch中实现多任务学习或知识蒸馏场景下的动态Loss权重调整，涵盖可学习标量权重、GradNorm算法及基于不确定性的加权方法，并解决设备一致性、计算图错误及权重约束问题。

Independent GitHub snapshot · Automated classification · Original documentation governs