mmcv/tests/test_cnn
Miao Zheng a4c3702628
add initializers and BaseModule for unified parameter initialization (#780)
* add initializers and BaseModule for unified parameter initialization

* fix circle import

* bug fix

* add is_init flag in BaseModule

* fix docstring

* sort import and fix doc format

* fix bug

* fix docformat and double quote string

* fix import sort

* import sort

* sort import

* revise according to comments

* fix doc format

* revise according to comments

* revise import and fix typo

* polish code

* revise minors

* revice minors

* revise apply function

* revise bias initialization with probability

* add type test for bias_prob

* revise minors
2021-02-07 11:13:57 +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 building ConvModule with HSigmoid using inplace=True and a… (#369) 2020-06-28 23:27:14 +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 Add utils to calculate model complexity and adjust folder structure for mmcv/cnn/utils . (#366) 2020-06-26 21:38:29 +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 Refactor plugins: move from ops to cnn (#380) 2020-07-03 01:37:58 +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_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_weight_init.py add initializers and BaseModule for unified parameter initialization (#780) 2021-02-07 11:13:57 +08:00
test_wrappers.py fix mmcv_ci test_wrappers.py for parrots (#758) 2020-12-24 19:50:46 +08:00