About BCEWithLogitsLoss's pos_weights

autograd
2

If your classes are imbalanced, setting the pos_weigth to number_neg_samples / number_pos_samples should give a more diverse prediction.