KLDivLoss can take a weight parameter but the docs don’t specify how it should be formatted. What should the format be?
It should be a 1D tensor having as many elements as you have classes. We’ll have to add that to the docs, thanks for reporting that!
Can it? I can see nothing weight related in the docs, nor in the source.