pytorch focal loss

focal loss的改进1：

Focal Loss改进版 GFocal Loss_jacke121的专栏-CSDN博客_focal loss改进

focal loss的改进2：

GHM loss

讲解：

5分钟理解Focal Loss与GHM——解决样本不平衡利器 - 知乎

代码：

import torch
import torch.nn as nn
import torch.nn.functional as F

class GHMC(nn.Module):
    def __init__(self,
                 bins: int=10,
                 reduction: str="mean",
                 ignore_index=-100):
        """
        计算改进版的FocalLoss
        :param bins: 表示将概率范围进行划分
        :param reduction: str型，表示loss最终累计的方式
        :param ignore_index: 表示忽略的index
        """
        super(GHMC, self).__init__()