mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmcv/tests/test_cnn
History
Shanghua Gao 6e3827d02b
[Feature] Support receptive field search of CNN models (#2056) 
* support rfsearch

* add labs for rfsearch

* format

* format

* add docstring and type hints

* clean code

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

* rm unused func

* update code

* update code

* update code

* update  details

* fix details

* support asymmetric kernel

* support asymmetric kernel

* Apply suggestions from code review

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

* Apply suggestions from code review

* Apply suggestions from code review

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

* Apply suggestions from code review

* Apply suggestions from code review

* Apply suggestions from code review

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

* Apply suggestions from code review

* add unit tests for rfsearch

* set device for Conv2dRFSearchOp

* Apply suggestions from code review

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

* remove unused function search_estimate_only

* move unit tests

* Update tests/test_cnn/test_rfsearch/test_operator.py

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

* Apply suggestions from code review

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

* Update mmcv/cnn/rfsearch/operator.py

Co-authored-by: Yue Zhou <592267829@qq.com>

* change logger

* Update mmcv/cnn/rfsearch/operator.py

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

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: lzyhha <819814373@qq.com>
Co-authored-by: Zhongyu Li <44114862+lzyhha@users.noreply.github.com>
Co-authored-by: Yue Zhou <592267829@qq.com>
 		2022-11-22 19:15:55 +08:00
..
test_rfsearch [Feature] Support receptive field search of CNN models (#2056) 2022-11-22 19:15:55 +08:00
test_build_layers.py [Fix] num_channels can not be divisible by num_groups in test_build_norm_layer (#2085) 2022-06-30 19:28:34 +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 Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +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 Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +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_model_registry.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_revert_syncbn.py [Fix] Fix parrots unit tests (#1913) 2022-05-09 16:59:32 +08:00
test_scale.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_silu.py [Enhancement] Support SiLU with torch < 1.7.0 (#2278) 2022-09-29 15:19:41 +08:00
test_swish.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_transformer.py Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +08:00
test_weight_init.py [Fix] Fix parrots unit tests (#1913) 2022-05-09 16:59:32 +08:00
test_wrappers.py Upgrade pre commit hooks (#2321) 2022-10-08 11:48:44 +08:00