Shortcuts

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:
  • 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