Hello Rishav!



 Rishav_Sapahia:

…
I am trying to implement custom weight maps to counter class imbalances.
…
I am really not sure how to incorporate custom weights in cross entropy loss .


Does the weight argument described in the documentation 49 for
CrossEntropyLoss do what you need?
Best.
K. Frank