loss-function - 超腾开源标签

本文介绍了PyTorch中常用的损失函数，按任务类型分类讲解。回归任务包括MSELoss（均方误差）、L1Loss（平均绝对误差）、SmoothL1Loss和HuberLoss；分类任务涵盖CrossEntropyLoss（交叉熵）、BCEWithLogitsLoss（推荐用于二分类）、NLLLoss及KLDivLoss等；特殊任务如CTCLoss用于序列对齐，TripletMarginLoss用于度量学习。文章还给出了任务与损失函数的匹配建议，并提醒注意输入张量形状、是否内置softmax以及正负样本平衡问题。