d6b449edcbdaf559807aaff6bd3234f47134e303,pyannote/audio/embedding/approaches/center_loss.py,CenterLoss,loss_and_grad,#CenterLoss#Any#Any#,206

Before Change

        X = batch["X"]
        y = batch["y"]

        fX = embed(X)

        func = value_and_multigrad(self.loss, argnums=[0, 1])

        loss, (fX_grad, fC_grad) = func(fX, self.fC_, y)

After Change

            fX = self.embed(embedding, batch["X"], internal=True)
            func = value_and_multigrad(self.loss_z, argnums=[0, 1])
            loss, (fX_grad, fC_grad) = func(fX, self.fC_, batch["y"],
                                            batch["n"])
            fX_grad = fX_grad[:, 0, :]

        else:

Instances