focal loss原理及实现

目录

原理

不含权重的focal loss

 带权重的focal loss

---

原理

不催不更

不含权重的focal loss

基于交叉熵的简单实现，学的时候先看这个

def focal_loss(logits, labels, gamma=5, reduction="mean"):
    # 经测试，reduction 加不加结果都一样，但是为了保险，还是加上
    ce_loss = F.cross_entropy(logits, labels, reduction="none") # 交叉熵
    log_pt