Shanghua Gao
|
bc10b5f0c0
|
[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>
[Fix] Fix skip_layer for RF-Next (#2489)
* judge skip_layer by fullname
* lint
* skip_layer first
* update unit test
|
2023-03-20 18:26:52 +08:00 |
Mashiro