PyTorch Forums
Passing the weights to CrossEntropyLoss correctly