mmclassification/tests/test_metrics
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
..
test_losses.py [Feature] Support classwise weight in losses (#388) 2021-08-31 10:44:12 +08:00
test_metrics.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00