Weights in BCEWithLogitsLoss

Yeah sorry for the typo. The t_n is missing indeed!

I think you are referring to the docs which states:

Must be a vector with length equal to the number of classes.

Now I understand the issue and am confused as well, as I thought a scalar tensor would work, e.h. pos_weight = num_neg / num_pos. Thanks for digging deeper!

CC @velikodniy who implemented this. Am I misunderstanding the usage of pos_weight?

1 Like