pytorch几种损失函数CrossEntropyLoss、NLLLoss、BCELoss、BCEWithLogitsLoss、focal_loss、heatmap_loss
分类问题常用的几种损失,记录下来备忘,后续不断完善。nn.CrossEntropyLoss()交叉熵损失常用于多分类问题CE=nn.CrossEntropyLoss()loss=CE(input,target)Input:(N,C),dtype:float,N是样本数量,在批次计算时通常就是batch_sizetarget:(N),dtype:long,是类别号,0≤targets[i]≤C−1p