[Fix] Add avg_non_ignore in cross entropy loss (#1409)
* [Fix] Add avg_non_ignore in cross entropy loss * [Fix] Add avg_non_ignore in cross entropy loss * add docstring * fix ut * fix docstring and comments * fix * fix bce * fix avg_factor in BCE and add more ut * add avg_non_ignore * add more ut * fix part of ut * fix part of ut * test avg_non_ignore would not affect ce/bce when reduction none/sum * test avg_non_ignore would not affect ce/bce when reduction none/sum/mean * re-organize ut * re-organize ut * re-organize ut * re-organize hardcode case * fix parts of comments * fix another parts of comments * fix
Showing
- docs/en/tutorials/training_tricks.md 20 additions, 0 deletionsdocs/en/tutorials/training_tricks.md
- docs/zh_cn/tutorials/training_tricks.md 25 additions, 0 deletionsdocs/zh_cn/tutorials/training_tricks.md
- mmseg/models/losses/cross_entropy_loss.py 80 additions, 12 deletionsmmseg/models/losses/cross_entropy_loss.py
- mmseg/models/losses/utils.py 5 additions, 1 deletionmmseg/models/losses/utils.py
- tests/test_models/test_losses/test_ce_loss.py 159 additions, 23 deletionstests/test_models/test_losses/test_ce_loss.py
Loading
Please register or sign in to comment