mmpretrain/mmcls/models/losses
LXXXXR 194ab7efda
[Feature] Add bce loss for multilabel task (#130)
* add bce loss for multilabel task

* minor change

* apply class wise sum

* fix docstring

* do not apply sum over classes and fix docstring

* fix docstring

* fix weight shape

* fix weight shape

* fix docstring

* fix linting issue

Co-authored-by: Y. Xiong <xiongyuxy@gmail.com>
2021-01-11 11:05:24 +08:00
..
__init__.py [Feature] Add bce loss for multilabel task (#130) 2021-01-11 11:05:24 +08:00
accuracy.py Update accuracy.py (#104) 2020-12-01 14:25:18 +08:00
cross_entropy_loss.py [Feature] Add bce loss for multilabel task (#130) 2021-01-11 11:05:24 +08:00
eval_metrics.py fix bug in eval_metrics (#122) 2020-12-23 16:20:47 +08:00
focal_loss.py [Feature] Add focal loss for multilabel task (#131) 2021-01-08 20:44:23 +08:00
label_smooth_loss.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
utils.py add weighted_loss 2020-07-12 00:06:55 +08:00