PyTorch Forums
Weights in BCEWithLogitsLoss