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 |