Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

M

mmsegmentation
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Environments
- Terraform modules
- Incidents

Snippets Groups Projects

ffcdfd36

Commit ffcdfd36 authored 4 years ago by Jerry Jiarui XU Committed by GitHub 4 years ago

Downloads
- Patches
- Plain Diff

[Enhancement] Support ignore_index for sigmoid BCE (#210)

* [Enhancement] Add args check for ignore_index

* Support ignore_index

parent 16fae6ef

No related branches found

No related tags found

Loading

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

configs/_base_/models/fast_scnn.py 3 additions, 3 deletions

configs/_base_/models/fast_scnn.py
configs/fastscnn/fast_scnn_4x8_80k_lr0.12_cityscapes.py 1 addition, 1 deletion

configs/fastscnn/fast_scnn_4x8_80k_lr0.12_cityscapes.py
mmseg/models/decode_heads/decode_head.py 2 additions, 1 deletion

mmseg/models/decode_heads/decode_head.py
mmseg/models/losses/cross_entropy_loss.py 31 additions, 12 deletions

mmseg/models/losses/cross_entropy_loss.py
tests/test_models/test_losses.py 11 additions, 1 deletion

tests/test_models/test_losses.py

with 48 additions and 18 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment