mwptoolkit.loss.nll_loss¶
- class mwptoolkit.loss.nll_loss.NLLLoss(weight=None, mask=None, size_average=True)[source]¶
Bases:
AbstractLoss
- Parameters
weight (Tensor, optional) – a manual rescaling weight given to each class.
mask (Tensor, optional) – index of classes to rescale weight