You are reading the documentation for MMClassification 0.x, which will soon be deprecated at the end of 2022. We recommend you upgrade to MMClassification 1.0 to enjoy fruitful new features and better performance brought by OpenMMLab 2.0. Check the installation tutorial, migration tutorial and changelog for more details.
- class mmcls.models.SeesawLoss(use_sigmoid=False, p=0.8, q=2.0, num_classes=1000, eps=0.01, reduction='mean', loss_weight=1.0)¶
Implementation of seesaw loss.
use_sigmoid (bool) – Whether the prediction uses sigmoid of softmax. Only False is supported. Defaults to False.
p (float) – The
pin the mitigation factor. Defaults to 0.8.
q (float) – The
qin the compenstation factor. Defaults to 2.0.
num_classes (int) – The number of classes. Default to 1000 for the ImageNet dataset.
eps (float) – The minimal value of divisor to smooth the computation of compensation factor, default to 1e-2.
reduction (str) – The method that reduces the loss to a scalar. Options are “none”, “mean” and “sum”. Default to “mean”.
loss_weight (float) – The weight of the loss. Defaults to 1.0
- forward(cls_score, labels, weight=None, avg_factor=None, reduction_override=None)¶
cls_score (torch.Tensor) – The prediction with shape (N, C).
labels (torch.Tensor) – The learning label of the prediction.
weight (torch.Tensor, optional) – Sample-wise loss weight.
avg_factor (int, optional) – Average factor that is used to average the loss. Defaults to None.
reduction (str, optional) – The method used to reduce the loss. Options are “none”, “mean” and “sum”.
The calculated loss
- Return type