mmclassification/mmcls
Ezra-Yu 192b79eea0
[Feature] Support classwise weight in losses (#388)
* Add classwise weight in losses:CE,BCE,softBCE

* Update unit test

* rm some extra code

* rm some extra code

* fix broadcast

* fix broadcast

* update unit tests

* use new_tensor

* fix lint
2021-08-31 10:44:12 +08:00
..
apis [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
core [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
datasets [Enhance] Add `hparams` argument in `AutoAugment` and `RandAugment` and some other improvement. (#398) 2021-08-24 18:15:54 +08:00
models [Feature] Support classwise weight in losses (#388) 2021-08-31 10:44:12 +08:00
utils [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
__init__.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
version.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00