[ torch 参数更多 ] torch.utils.data.SubsetRandomSampler

torch.utils.data.SubsetRandomSampler

torch.utils.data.SubsetRandomSampler(indices, generator=None)

paddle.io.SubsetRandomSampler

paddle.io.SubsetRandomSampler(indices)

两者参数完全一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
indices indices 子集在原数据集中的索引序列,需要是 list 或者 tuple 类型。
generator - Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。