我们从Python开源项目中,提取了以下12个代码示例,用于说明如何使用torch.nn.NLLLoss2d()。
def __init__(self, weight=None): super(CrossEntropyLoss2d,self).__init__() self.loss = nn.NLLLoss2d(weight)
def log_loss(loss_weight=None, size_ave=True, dim=2): if dim == 1: return nn.NLLLoss(weight=loss_weight,size_average=size_ave) elif dim == 2: return nn.NLLLoss2d(weight=loss_weight,size_average=size_ave)
def __init__(self, weight=None, size_average=True, ignore_index=255): super(CrossEntropyLoss2d, self).__init__() self.nll_loss = nn.NLLLoss2d(weight, size_average, ignore_index)
def __init__(self, gamma=2, weight=None, size_average=True, ignore_index=255): super(FocalLoss2d, self).__init__() self.gamma = gamma self.nll_loss = nn.NLLLoss2d(weight, size_average, ignore_index)
def __init__(self, size_average=True): super(CrossEntropyLoss2D, self).__init__() self.nll_loss_2d = nn.NLLLoss2d(size_average=size_average)
def __init__(self, weights, size_average=True): super(MSCrossEntropyLoss2D, self).__init__() self.nll_loss_2d = nn.NLLLoss2d(size_average=size_average) self.weights = weights
def __init__(self, weight=None): super().__init__() self.loss = nn.NLLLoss2d(weight)