InfiniteSampler¶
- class mmengine.dataset.InfiniteSampler(dataset, shuffle=True, seed=None)[source]¶
It’s designed for iteration-based runner and yields a mini-batch indices each time.
The implementation logic is referred to https://github.com/facebookresearch/detectron2/blob/main/detectron2/data/samplers/distributed_sampler.py
- Parameters: