mmpretrain/mmcls/models
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
..
backbones Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
classifiers Visualize results on image demo (#58) 2020-10-10 16:33:27 +08:00
heads Fix bug and optimize MNIST config (#98) 2020-11-26 15:27:04 +08:00
losses [Feature] Add bce loss for multilabel task (#130) 2021-01-11 11:05:24 +08:00
necks Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
utils Dev mobilenetv3 2020-06-30 15:50:36 +08:00
__init__.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
builder.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00