mmsegmentation

Commit Graph

Author	SHA1	Message	Date
谢昕辰	b379b5a5b3	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
Jerry Jiarui XU	3150dd0ce4	refactor test organization (#440 ) * refactor test organization * fixed se layer * update mmcv uper bound	2021-03-30 17:55:09 -07:00

Author

SHA1

Message

Date

谢昕辰

b379b5a5b3

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

Jerry Jiarui XU

3150dd0ce4

refactor test organization (#440 )

* refactor test organization

* fixed se layer

* update mmcv uper bound

2021-03-30 17:55:09 -07:00

2 Commits (e5007e7491952bc4abbe93902882b70ba9d0b98e)