Solved by ptrblck in post #4 
              
              
                Good to hear. 
Maybe the new pos_weight argument for nn.BCEwithLogitsLoss might work better, since there is a difference in my weighting approach and the implemented one. See this discussion.