f0fe1d57b54b7670ad4586e19d4383db561e84e7,code/deep/DDC_DeepCoral/mmd.py,MMD_loss,linear_mmd2,#MMD_loss#,32

Before Change

    def linear_mmd2(self, f_of_X, f_of_Y):
        loss = 0.0
        delta = f_of_X - f_of_Y
        loss = torch.mean((delta[:-1] * delta[1:]).sum(1))
        return loss

    def forward(self, source, target):

After Change

    def linear_mmd2(self, f_of_X, f_of_Y):
        loss = 0.0
        delta = f_of_X.float().mean() - f_of_Y.float().mean()
        loss = delta.dot(delta.T)
        return loss

    def forward(self, source, target):
        if self.kernel_type == "linear":

Instances