ColossalAI/colossalai/legacy/utils/data_sampler
..
__init__.py
base_sampler.py
data_parallel_sampler.py