[Feature] Support multiple losses during training (#818)
* multiple losses
* fix lint error
* fix typos
* fix typos
* Adding Attribute
* Fixing loss_ prefix
* Fixing loss_ prefix
* Fixing loss_ prefix
* Add Same
* loss_name must has 'loss_' prefix
* Fix unittest
* Fix unittest
* Fix unittest
* Update mmseg/models/decode_heads/decode_head.py
Co-authored-by:
Junjun2016 <hejunjun@sjtu.edu.cn>
Showing
- docs/tutorials/training_tricks.md 18 additions, 0 deletionsdocs/tutorials/training_tricks.md
- docs_zh-CN/tutorials/training_tricks.md 19 additions, 0 deletionsdocs_zh-CN/tutorials/training_tricks.md
- mmseg/core/seg/sampler/ohem_pixel_sampler.py 8 additions, 6 deletionsmmseg/core/seg/sampler/ohem_pixel_sampler.py
- mmseg/models/decode_heads/decode_head.py 35 additions, 8 deletionsmmseg/models/decode_heads/decode_head.py
- mmseg/models/decode_heads/point_head.py 3 additions, 2 deletionsmmseg/models/decode_heads/point_head.py
- mmseg/models/losses/cross_entropy_loss.py 20 additions, 1 deletionmmseg/models/losses/cross_entropy_loss.py
- mmseg/models/losses/dice_loss.py 19 additions, 0 deletionsmmseg/models/losses/dice_loss.py
- mmseg/models/losses/lovasz_loss.py 20 additions, 1 deletionmmseg/models/losses/lovasz_loss.py
- tests/test_models/test_heads/test_decode_head.py 89 additions, 0 deletionstests/test_models/test_heads/test_decode_head.py
- tests/test_models/test_losses/test_ce_loss.py 14 additions, 3 deletionstests/test_models/test_losses/test_ce_loss.py
- tests/test_models/test_losses/test_dice_loss.py 20 additions, 4 deletionstests/test_models/test_losses/test_dice_loss.py
- tests/test_models/test_losses/test_lovasz_loss.py 32 additions, 8 deletionstests/test_models/test_losses/test_lovasz_loss.py
Loading
Please register or sign in to comment