Passing the weights to CrossEntropyLoss correctly - PyTorch ...
discuss.pytorch.org › t › passing-the-weights-toMar 10, 2018 · Hi, I just wanted to ask how the mechanism of passing the weights to CrossEntropyLoss works. Currently, I have a list of class labels that are [0, 1, 2, 3, 4, 5, 6, 7 ...