megengine.data.SequentialSampler¶
- class SequentialSampler(dataset, batch_size=1, drop_last=False, indices=None, world_size=None, rank=None)[源代码]¶
按序列顺序对元素进行采样。
- 参数
dataset – 待采样的目标数据集。
batch_size – 使用 batch 方法时指定 batch 大小。
drop_last – 如果batch大小不能整除数据集大小时，为
True则放弃最后一个不完整的batch; 为False则最后一个batch可能比较小。默认：Falseindices – indice of samples.
world_size – 进程数。
rank – 进程号，在0与
world_size范围内的非负整数。
方法
batch()Batch 方法提供了一个 batch 索引生成器。
sample()返回一个生成器。
scatter(indices)分散方法用于将下标拆分为子集，每个子集将分配一个编号。