Hello, author, in the cov function of loss_reweighting.py. I think there is a little different with the definition in the paper:
Your code: cov = torch.matmul((w * x).t(), x)
Your paper:

Why the code is not cov = torch.matmul((w * x).t(), w * x)
Thank you very much.