欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

SmoothL1Loss的初步学习理解

程序员文章站 2023-12-31 22:15:22
...

主要参考了这一篇和这一篇博客

另外附上L1损失函数代码:

def _smooth_l1_loss(x, t, in_weight, sigma):        #sigma=1
    sigma2 = sigma ** 2
    diff = in_weight * (x - t)
    abs_diff = diff.abs()
    flag = (abs_diff.data < (1. / sigma2)).float()
    y = (flag * (sigma2 / 2.) * (diff ** 2) +
         (1 - flag) * (abs_diff - 0.5 / sigma2))
    return y.sum()

上一篇:

下一篇: