mmcls.models.MultiLabelClsHead

class mmcls.models.MultiLabelClsHead(loss={'loss_weight': 1.0, 'reduction': 'mean', 'type': 'CrossEntropyLoss', 'use_sigmoid': True}, init_cfg=None)

Classification head for multilabel task.

Parameters

loss (dict) – Config of classification loss.