Custom BCE loss with pos and neg weights

Hi,

What do you want to accomplish with this?
Also, could you please specify the variables a_i and x_i you’re using?

This might be worth checking out to make sure that you actually need the second set of weights About BCEWithLogitsLoss’s pos_weights.