Closed zwishenzug closed 1 year ago
I got an error about dot not being implemented for half. I had to change the code in weight_matching to this to get it to work:
oldL = torch.vdot(torch.flatten(A).float(), torch.flatten(torch.eye(n)[perm[p].long()]).float()).half() newL = torch.vdot(torch.flatten(A).float(), torch.flatten(torch.eye(n)[ci, :]).float()).half()
I got an error about dot not being implemented for half. I had to change the code in weight_matching to this to get it to work: