mmcv/tests/test_cnn
Shilong Zhang e05fb56031
Refactor the baseclass related to transformer (#978)
* minor changes

* change to modulist

* change to Sequential

* replace dropout with attn_drop and proj_drop in MultiheadAttention

* add operation_name for attn

* add drop path and move all ffn args to ffncfgs

* fix typo

* fix a bug when use default value of ffn_cfgs

* fix ffns

* add deprecate warning

* fix deprecate warning

* change to pop kwargs

* support register FFN of transformer

* support batch first

* fix batch first wapper

* fix forward wapper

* fix typo

* fix lint

* add unitest for transformer

* fix unitest

* fix equal

* use allclose

* fix comments

* fix comments

* change configdict to dict

* move drop to a file

* add comments for drop path

* add noqa 501

* move bnc wapper to MultiheadAttention

* move bnc wapper to MultiheadAttention

* use dep warning

* resolve comments

* add unitest:

* rename residual to identity

* revert runner

* msda residual to identity

* rename inp_identity to identity

* fix name

* fix transformer

* remove key in msda

* remove assert for key

Co-authored-by: HIT-cwh <2892770585@qq.com>
Co-authored-by: bkhuang <congee524@gmail.com>
Co-authored-by: Wenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
2021-06-11 18:09:31 +08:00
..
test_build_layers.py [Bug] Avoid recursion error when using customized norm layer without _abbr_ (#732) 2020-12-22 10:19:52 +08:00
test_context_block.py Refactor plugins: move from ops to cnn (#380) 2020-07-03 01:37:58 +08:00
test_conv2d_adaptive_padding.py Implementation of 2D convolution in tensorflow with `padding` as "same" (#529) 2020-09-09 23:57:48 +08:00
test_conv_module.py fix bug of convmodule (#889) 2021-03-20 23:08:20 +08:00
test_depthwise_seperable_conv_module.py Add Depthwise Seperable ConvModule (#477) 2020-08-09 17:29:02 +08:00
test_flops_counter.py [Feature] support to calculate FLOPs of GN, IN, LN (#897) 2021-03-19 14:04:49 +08:00
test_fuse_conv_bn.py Move fuse conv bn to mmcv (#382) 2020-07-08 19:03:15 +08:00
test_generalized_attention.py [Fix] fix generalized attention fp16 (#1036) 2021-05-23 15:16:27 +08:00
test_hsigmoid.py [Enhance]enhance hsigmoid (#657) 2020-11-19 21:26:26 +08:00
test_hswish.py add ext ops, support parrots (#310) 2020-06-28 23:15:47 +08:00
test_model_registry.py add model registry (#760) 2021-04-10 21:33:34 +08:00
test_non_local.py Add pairwise function for 'gaussian' and 'concatenation' mode in NonLocal. (#424) 2020-07-17 23:44:50 +08:00
test_scale.py Add building bricks of cnn (#247) 2020-05-01 00:32:25 +08:00
test_swish.py add Swish activation (#522) 2020-08-27 00:39:17 +08:00
test_transformer.py Refactor the baseclass related to transformer (#978) 2021-06-11 18:09:31 +08:00
test_weight_init.py [Feature] Support to use name of the base classes in init_cfg (#1057) 2021-06-01 22:35:19 +08:00
test_wrappers.py fix mmcv_ci test_wrappers.py for parrots (#758) 2020-12-24 19:50:46 +08:00