How to weight the loss?

5

Thanks!
but weights.sum() is correct? not the size of batch?