Good to hear.
Maybe the new pos_weight argument for nn.BCEwithLogitsLoss 1.3k might work better, since there is a difference in my weighting approach and the implemented one. See this discussion 1.2k.