InfiniteSampler

class mmengine.dataset.InfiniteSampler(dataset, shuffle=True, seed=None)

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

参数

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

返回类型

None

set_epoch(epoch)

Not supported in iteration-based runner.

参数

epoch (int) –

返回类型

None