PaddleOCR2.7.5 训练到几百回合出现loss>1问题

[18325391772]

报错代码： class DiceLoss(nn.Layer): def init(self, eps=1e-6): super(DiceLoss, self).init() self.eps = eps

def forward(self, pred, gt, mask, weights=None):
    """
    DiceLoss function.
    """

    assert pred.shape == gt.shape
    assert pred.shape == mask.shape
    if weights is not None:
        assert weights.shape == mask.shape
        mask = weights * mask
    intersection = paddle.sum(pred * gt * mask)

    union = paddle.sum(pred * mask) + paddle.sum(gt * mask) + self.eps
    loss = 1 - 2.0 * intersection / union
    assert loss <= 1
    return loss

[GreatV]

升级到2.8.0试试，照理说不会出现loss>1的情况