ReplacementSampler#

class ReplacementSampler(dataset, batch_size=1, drop_last=False, num_samples=None, weights=None, world_size=None, rank=None, seed=None)[源代码]#

有放回(可重复选取)地进行随机采样。

参数:
  • dataset – 待采样的目标数据集。

  • batch_size – 使用 batch 方法时指定 batch 大小。

  • drop_last – 如果batch大小不能整除数据集大小时,为 True 则放弃最后一个不完整的batch; 为 False 则最后一个batch可能比较小。默认:False

  • num_samples – 分配给一个进程的样本数。

  • weights – 样本索引的权重,可以是未归一化的权重。

  • world_size – 进程数。

  • rank – 进程号,在0与 world_size 范围内的非负整数。

  • seed – 随机种子。

sample()[源代码]#

返回一个列表,包含了所有样本的索引。

返回类型:

List