mmcv/mmcv/cnn
takuoko d510b8b174
[Feature] Support LayerScale in FFN (#2451)
* add layer scale

* add layer scale

* add layer scale

* Update mmcv/cnn/bricks/transformer.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update mmcv/cnn/bricks/transformer.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* add layer scale

* move LayerScale

* add layer_scale_init_value

* add typehint

* fix for tensor with any dim

* fix layer scale rule

* fix layer scale rule

* fix test

* add docs

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-11 17:58:19 +08:00
..
bricks [Feature] Support LayerScale in FFN (#2451) 2022-12-11 17:58:19 +08:00
utils resolve conflicts 2022-10-22 13:39:50 +08:00
__init__.py [Refactor] Use MODELS registry in mmengine and delete basemodule (#2172) 2022-08-19 11:18:47 +08:00
alexnet.py Remove runner, parallel, engine and device (#2216) 2022-08-22 19:16:30 +08:00
resnet.py [Refactor] Modify the importing path due to mmengine changes (#2224) 2022-08-24 19:48:10 +08:00
vgg.py [Refactor] Modify the importing path due to mmengine changes (#2224) 2022-08-24 19:48:10 +08:00