ok so if i am using nn.BCEWithLogitsLoss, that means i only have one class output. does that mean i only have to pass only value for the pos_weight argument ? and is there a general formula to find out how much weight to give to each class ?