图像分类:标签平滑损失函数,用来防止过拟合
标签平滑损失函数:
class LabelSmoothingCrossEntropy(nn.Module):
def __init__(self, epsilon: float = 0.1, reduction='mean'):
super().__init__()
self.epsilon, self.reduction = epsilon, reduction
def forward(self, output, target):
原创
2021-09-10 11:11:23 ·
1030 阅读 ·
0 评论