search

jiequancui / DKL

Decoupled Kullback-Leibler Divergence Loss (DKL)
https://arxiv.org/pdf/2305.13948v1.pdf
21 stars 3 forks source link

weighted Mean Square Error (wMSE) loss #2

Open HajarMazaheri opened 1 month ago

HajarMazaheri commented 1 month ago

Hi. Can we use nn.MSELoss() to calculate (wMSE)? Thank you for guiding me

jiequancui commented 1 month ago

Hi,

Thanks for your interest in our work.

I think it is equivalent to implement it directly by torch.nn.MSELoss.

Best, Jiequan