INNER CODE UNIT · Python

dist

KaiyangZhou/pytorch-center-loss · center_loss.py:41

        dist = distmat * mask.float()
        loss = dist.clamp(min=1e-12, max=1e+12).sum() / batch_size

        return loss

View source record →