mmcv/tests/test_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
..
test_build_layers.py [Refactor] Modify the importing path due to mmengine changes (#2224) 2022-08-24 19:48:10 +08:00
test_context_block.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_conv2d_adaptive_padding.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_conv_module.py [Refactor] Modify the importing path due to mmengine changes (#2224) 2022-08-24 19:48:10 +08:00
test_depthwise_seperable_conv_module.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_flops_counter.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_fuse_conv_bn.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_generalized_attention.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_hsigmoid.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_hswish.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_non_local.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_scale.py [Feature] Support LayerScale in FFN (#2451) 2022-12-11 17:58:19 +08:00
test_silu.py [Enhancement] Support SiLU with torch < 1.7.0 2022-10-22 13:39:50 +08:00
test_swish.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_transformer.py [Feature] Support LayerScale in FFN (#2451) 2022-12-11 17:58:19 +08:00
test_wrappers.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00