mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmsegmentation/tests/test_models
History
sennnnn c27ef91942 Adjust vision transformer backbone architectures (#524) 
* Adjust vision transformer backbone architectures;

* Add DropPath, trunc_normal_ for VisionTransformer implementation;

* Add class token buring intermediate period and remove it during final period;

* Fix some parameters loss bug;

* * Store intermediate token features and impose no processes on them;

* Remove class token and reshape entire token feature from NLC to NCHW;

* Fix some doc error

* Add a arg for VisionTransformer backbone to control if input class token into transformer;

* Add stochastic depth decay rule for DropPath;

* * Fix output bug when input_cls_token=False;

* Add related unit test;

* * Add arg: out_indices to control model output;

* Add unit test for DropPath;

* Apply suggestions from code review

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
 		2021-04-30 10:37:47 -07:00
..
test_backbones Adjust vision transformer backbone architectures (#524) 2021-04-30 10:37:47 -07:00
test_heads refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_losses [Enhance] Support reading class_weight from file in loss functions to help MMDet3D (#513) 2021-04-29 01:04:15 -07:00
test_necks add upsample neck (#512) 2021-04-24 21:22:09 -07:00
test_segmentors refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_utils Adjust vision transformer backbone architectures (#524) 2021-04-30 10:37:47 -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