Weighted Loss

[parthnatekar]

Hi,

I can see an implementation for weighted MSE loss.

Can you explain what how you get the weights for this? I cannot find this in your paper either.

Thanks

[RitaHT]

Same question, I just use an external make_weight_map function as temporary makeup for this