mmsegmentation/tests/test_models
谢昕辰 5b33faa146
support transformer backbone (#465)
* vit backbone

* fix lint

* add docstrings and fix pretrained pos_embed dim not match prob

* add unittest for vit

* fix lint

* add vit based fcn configs

* fix import error

* support multiple resolution input images

* upsample pos_embed at init_weights

* support resize pos_embed at evaluation

* fix training errors

* add more unitest code for vit backbone

* unitest for uncovered code

* add norm_eval unittest

* refactor _pos_embeding

* minor change

* change var name

* rafactor init_weight

* load weights after resize

* ignore 'module' in pretrain checkpoint

* add with_cp

* add with_cp

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2021-04-21 20:19:55 -07:00
..
test_backbones support transformer backbone (#465) 2021-04-21 20:19:55 -07:00
test_heads refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_losses refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_necks refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_segmentors refactor test organization (#440) 2021-03-30 17:55:09 -07:00
__init__.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_forward.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00