mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmclassification/mmcls/models/losses
History
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
..
__init__.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
accuracy.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
asymmetric_loss.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
cross_entropy_loss.py [Feature] Support classwise weight in losses (#388) 2021-08-31 10:44:12 +08:00
focal_loss.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
label_smooth_loss.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
utils.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00