[Feature] Add data sampler (#30)
* Add DefaultSampler and InfiniteSampler * Add unit test * Add samplers to API reference * Update docstring * Improve according to comments * Rename `num_replicas` to `world_size` * Update docstring. * Update mmengine/data/sampler.py Co-authored-by:Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/data/sampler.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Fix typo in unit test Co-authored-by:
Wenwei Zhang <40779233+ZwwWayne@users.noreply.github.com> Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Showing
- docs/en/api.rst 5 additions, 0 deletionsdocs/en/api.rst
- docs/zh_cn/api.rst 5 additions, 0 deletionsdocs/zh_cn/api.rst
- mmengine/data/__init__.py 4 additions, 0 deletionsmmengine/data/__init__.py
- mmengine/data/sampler.py 168 additions, 0 deletionsmmengine/data/sampler.py
- tests/test_data/test_sampler.py 139 additions, 0 deletionstests/test_data/test_sampler.py
Loading
Please register or sign in to comment