Commit Graph

3 Commits (26a93f201cf73ded95fa6834ce4d48b83e0e9425)

Author SHA1 Message Date
Ziyi Wu 771ca7d3e0 [Enhance] Support reading class_weight from file in loss functions to help MMDet3D (#513)
* support reading class_weight from file in loss function

* add unit test of loss with class_weight from file

* minor fix

* move get_class_weight to utils
2021-04-29 01:04:15 -07:00
Jerry Jiarui XU ffcdfd36d0 [Enhancement] Support ignore_index for sigmoid BCE (#210)
* [Enhancement] Add args check for ignore_index

* Support ignore_index
2020-11-17 16:14:03 +08:00
Jiarui XU b2724da80b init commit 2020-07-10 02:39:01 +08:00