d2l-pytorch icon indicating copy to clipboard operation
d2l-pytorch copied to clipboard
verified publisher icon dsgiitr

fix: use alpha_t in FocalLoss as per standard formula

Open wt7k opened this issue 7 months ago • 1 comments

  • 之前 FocalLoss 中计算的 alpha_t 未被使用，导致权重没加上
  • 修改后 loss 计算遵循论文公式：loss = - alpha_t * (1 - pt)^gamma * log(pt)
  • 保留原先 pt 的计算方式，符合概率定义

wt7k avatar Jul 07 '25 13:07 wt7k