How to use class weights in loss function for imbalanced dataset

[Kathi (Katharina)]

from torch import nn

weights = [0.4, 1]
class_weights=torch.FloatTensor(weights).cuda()
learn.crit = nn.CrossEntropyLoss(weight=class_weights)