InfiniteSampler — mmengine 0.10.7 documentation (original) (raw)

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 tohttps://github.com/facebookresearch/detectron2/blob/main/detectron2/data/samplers/distributed_sampler.py

Parameters:

• dataset (Sized) – The dataset.
• shuffle (bool) – Whether shuffle the dataset or not. Defaults to True.
• seed (int, optional) – Random seed. If None, set a random seed. Defaults to None.

set_epoch(epoch)[source]¶

Not supported in iteration-based runner.

Parameters:

epoch (int) –

Return type:

None