mmcls.models.ClsHead¶ class mmcls.models.ClsHead(loss={'loss_weight': 1.0, 'type': 'CrossEntropyLoss'}, topk=(1,), cal_acc=False, init_cfg=None)[source]¶ classification head. Parameters loss (dict) – Config of classification loss. topk (int | tuple) – Top-k accuracy. cal_acc (bool) – Whether to calculate accuracy during training. If you use Mixup/CutMix or something like that during training, it is not reasonable to calculate accuracy. Defaults to False.