TorchJD支持多损失训练
TorchJD 项目分享了在 PyTorch 中进行多损失训练的进展,覆盖标量化与 Jacobian descent 两类方法。该工具面向多任务、约束、辅助损失等场景,尝试通过梯度聚合让每个损失项同时下降。
事件进展(1 篇报道)
-
2026-07-08 00:20Reddit r/MachineLearning TorchJD: Training with multiple losses in PyTorch [P]
TorchJD 项目分享了在 PyTorch 中进行多损失训练的进展,覆盖标量化与 Jacobian descent 两类方法。该工具面向多任务、约束、辅助损失等场景,尝试通过梯度聚合让每个损失项同时下降。