mmsegmentation/tests/test_models/test_backbones
Rockey 7a1c9a5499
[Fix] Fix the bug that vit cannot load pretrain properly when using i… (#999)
* [Fix] Fix the bug that vit cannot load pretrain properly when using init_cfg to specify the pretrain scheme

* [Fix] fix the coverage problem

* Update mmseg/models/backbones/vit.py

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>

* [Fix] make the predicate more concise and clearer

* [Fix] Modified the judgement logic

* Update tests/test_models/test_backbones/test_vit.py

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>

* add comments

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-11-03 10:36:09 -07:00
..
__init__.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_bisenetv1.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_bisenetv2.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_blocks.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_cgnet.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_fast_scnn.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_hrnet.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_icnet.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_mit.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_mobilenet_v3.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_resnest.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_resnet.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_resnext.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_swin.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_timm_backbone.py [Feature] Support TIMMBackbone (#998) 2021-11-02 12:51:11 +08:00
test_unet.py [Enhancement] Add codespell pre-commit hook and fix typos (#920) 2021-10-13 06:21:17 -07:00
test_vit.py [Fix] Fix the bug that vit cannot load pretrain properly when using i… (#999) 2021-11-03 10:36:09 -07:00
utils.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00