* [Feature] Add support for MPS
* fix import error
* update ut
* fix error
* trigger CI
* use a unique basename for test file modules
* avoid bc-breaking
* Fix
* Fix
* fix type hint
* minor fix
* remove some type hints of functions or methods
* minor fix
* Apply suggestions from code review
* minor fix
* minor refinement
Co-authored-by: HAOCHENYE <21724054@zju.edu.cn>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zhouzaida <zhouzaida@163.com>
* [Enhance] Add type hints in /ops:
`fused_bias_leakyrelu.py`, 'gather_points.py`, `group_points.py`.
There is no need to add type hints in `furthest_point_sample.py` and
`info.py`.
As for `focal_loss.py`, please see #1994.
* Modidied the default value of a variable.
* [Enhance] Add type hints in:
`knn.py`, `masked_conv.py`, `merge_cells.py`, `min_area_polygons.py`,
`modulated_deform_conv.py`, multi_scale_deform_attn.py`.
* Fix typehint.
* Fixed typehint.
* remove type hints of symbolic
* add no_type_check to ignore mypy check for method
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Add torch_meshgrid_ij wrapper due to PyTorch change
* Update torch_meshgrid name/doc/version implementation
* Make imports local
* add ut
* ignore ut when torch is not available
Co-authored-by: zhouzaida <zhouzaida@163.com>
* Fix batched_nms for rotated box.
Add type hint for nms.py
* Add test
* doc string
* revert symbolic hint
* fix max_coordinate
* add comment
* rename type
* fix typo docstring
* [Feature] Support Psamask with cambricon MLU backend
* [Fix] refine the format according to lint check
* Replace std:max/min with conditional operators
* [Fix] Format head files
* Add typehint in mmcv/cnn/utils/*
* Update mmcv/cnn/utils/flops_counter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/cnn/utils/weight_init.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Fix bugs
* Fix
* Fix2
* Update mmcv/cnn/utils/weight_init.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/cnn/utils/flops_counter.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/cnn/utils/weight_init.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/cnn/utils/weight_init.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Fix
* minor fix
* line is too long.
* Update mmcv/cnn/utils/weight_init.py
Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>
* Fix
* Update mmcv/cnn/utils/weight_init.py
* fix default value of float type hint
* fix default value of float type hint
* fix default value of float type hint
* fix default value of float type hint
* fix default value of float type hint
* fix default value of float type hint
* fix default value of float type hint
* Fix
* minor refinement
* replace list with tuple for input
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: zhouzaida <zhouzaida@163.com>
* Add type hints for mmcv/engine/test.py
* Add type hints for mmcv/engine/test.py
* Add type hints for mmcv/engine/test.py
* fix type hint
Co-authored-by: zhouzaida <zhouzaida@163.com>
* Add type hints for mmcv/ops/ball_query.py, border_align.py and correlation.py
* Add type hints for mmcv/ops/deform_conv.py, deform_roi_pool.py and deprecated_wrappers.py
* Remove type hints for deform_conv.py and deform_roi_pool.py
* Fix type hints for other files
* [Enhance] Add type ints in these files:
'base_module.py', 'checkpoint.py',
'default_constructor.py',
'dist_utils.py', 'fp16_utils.py',
'log_buffer.py', 'priority.py'.
* Fixed all the inappropriate type hints.
Removed the return type of __init__ funcs.
* Fixed type hint.
* Added type hints in `dist_utils.py` and `log_buffer.py`.
* Remove unnecessary import.
* minor fix
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* [Enhance] Add type hint in `focal_loss.py`.
* Use torch.LongTensor as type of target.
* Fixed the missing type hint.
* Removed some unnecessary type hint.
* Add missing type hint and fix the type hint of `target`.
* Fix the format of typehint.
* minor refinement
* minor fix
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Add type hints in resnet.py
* using lint while commit
* using lint while commit
* using lint while commit
* reslove typehints
* add pre-defined type hints
* Add type hints for other methods in mmcv/cnn/resnet.py
* Fix type hints
* Add type hints in mmcv/ops/carafe.py
* Add type hints in mmcv/ops/corner_pool.py
* Add type hints in mmcv/ops/diff_iou_rotated.py
* Add type hints for other methods for mmcv/ops/corner_pool.py
* Add type hints for other methods in mmcv/ops/carafe.py
* Add type hints for symbolic method
* [Enhance] Add type hint in 'utils.py'.
* [Enhance] Add type hint in 'default_constructor.py'.
* Fixed undefined name.
* minor refinement
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>