mmsegmentation/tests/test_models
MengzhangLI cc89c8d18d [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
2022-03-30 18:32:47 +08:00
..
test_backbones [Feature] Add BEiT backbone (#1404) 2022-03-30 15:25:10 +08:00
test_heads [Feature] Support K-Net (#1289) 2022-03-10 20:05:37 +08:00
test_losses [Fix] Add avg_non_ignore in cross entropy loss (#1409) 2022-03-30 18:32:47 +08:00
test_necks [Feature] Add BEiT backbone (#1404) 2022-03-30 15:25:10 +08:00
test_segmentors [Fix] Fix loss parse in val_step (#906) 2021-09-25 18:17:40 -07:00
test_utils [Feature] add nlc2nchw2nlc and nchw2nlc2nchw (#1249) 2022-03-10 20:27:28 +08:00
__init__.py
test_forward.py [Fix] Convert SyncBN to BN when training on DP (#772) 2021-09-15 09:39:37 -07:00